2024华为OD机试D卷 - 攀登者2

/***
 * 攀登者喜欢寻找各种地图，并且尝试攀登到最高的山峰。
 *
 * 地图表示为一维数组，数组的索引代表水平位置，数组的元素代表相对海拔高度。其中数组元素0代表地面。
 *
 * 例如：[0,1,2,4,3,1,0,0,1,2,3,1,2,1,0]，代表如下图所示的地图，地图中有两个山脉位置分别为 1,2,3,4,5 和 8,9,10,11,12,13，最高峰高度分别为 4,3。最高峰位置分别为3,10。
 *
 * 一个山脉可能有多座山峰(高度大于相邻位置的高度，或在地图边界且高度大于相邻的高度)。
 *
 * 作者：华为OD机试题库
 * 链接：https://www.nowcoder.com/discuss/637024593848320000?sourceSSR=users
 * 来源：牛客网
 */

public class Num001 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int[] heights = Arrays.stream(scanner.nextLine().split(","))
                .mapToInt(Integer::parseInt)
                .toArray();
        int strength = Integer.parseInt(scanner.nextLine());
        int num = getNum(heights, strength);
        System.out.println(num);
    }

    public static int getNum(int[] nums, int strength) {
        int ans = 0;
        for (int i = 0; i < nums.length; i++) {
if (isPeak(nums, i) && strength > nums[i] * 3) {
                ans++;
            }
        }
        return ans;
    }

    public static Boolean isPeak(int[] nums, int i) {
        if (i + 1 < nums.length && i - 1 >= 0 && nums[i] > nums[i - 1] && nums[i] > nums[i + 1])
            return true;
        else return false;
    }
}