求解单峰数组

【问题描述】

一个整数数组被称为单峰数组，当

• 第一部分为严格递增序列

• 中间部分为常数序列

• 最后部分为严格递减序列

• 第一部分和最后部分都可以缺失

• 例如，下列3个数组的单峰的：[5, 7, 11, 11, 2, 1]、[4, 4, 2]、[7]，但是以下3个数组不是单峰的：[5, 5, 6, 6, 1]、[1, 2, 1, 2]、[4, 5, 5, 6]。

• 编写程序判断一个数组是否为单峰的。

【输入形式】

输入的第一行为一个整数T，表示测试用例的个数。

接下来的2T行，每两行对应一个测试用例，每个用例的第一行为一个正整数n，表示数组中的元素个数，接下来一行为n个整数，表示数组的n个元素

【输出形式】

输出有T行，每行对应一个测试用例的结果，如果为单峰数组，输出“YES”，否则输出“NO”。

【样例输入】

3
6
1 5 5 5 4 2
5
10 20 30 20 10
4
1 2 1 2

【样例输出】

YES
YES
NO

解题思路

1. 初始化状态：定义三个状态来表示当前遍历的阶段——递增、平坦、递减。最开始，我们处于递增阶段。

2. 遍历数组：从头到尾遍历数组，根据当前元素与下一个元素的关系，决定是否需要改变状态。

   1. 如果当前状态为递增，且下一个元素大于当前元素，保持递增状态。

   2. 如果当前状态为递增，且下一个元素等于当前元素，切换到平坦状态。

   3. 如果当前状态为平坦，且下一个元素等于当前元素，保持平坦状态。

   4. 如果当前状态为平坦或递增，且下一个元素小于当前元素，切换到递减状态。

   5. 如果当前状态为递减，且下一个元素小于当前元素，保持递减状态。

3. 状态违规检查：在遍历过程中，如果发现状态的变化不符合单峰数组的定义（如从递减切换到递增或平坦，或从递减切换到平坦），则立即判断数组不是单峰的。

4. 遍历完成：如果能够顺利完成遍历，且状态变化符合单峰数组的定义，则数组是单峰的。

Java代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int T = scanner.nextInt();//测试用例个数
        for(int i = 0; i < T; i++){
            int n = scanner.nextInt();//数组元素的个数
            int[] arr = new int[n];
            for (int j = 0; j < n; j++) {
                arr[j] = scanner.nextInt();//数组元素
            }
            System.out.println(isSinglePeak(arr) ? "YES" : "NO");
        }
    }
    private static boolean isSinglePeak(int[] arr){
        int phase = 0; //0:递增；1：平坦；2：递减
        for (int i = 0; i < arr.length-1; i++) {
            if(arr[i] < arr[i+1]){
                if(phase > 0) return false; // 如果已经不是递增阶段，则不符合单峰
            }else if (arr[i] == arr[i+1]){
                if (phase == 2) return false; // 如果已经是递减阶段，不能再平坦
                phase = 1; // 进入或保持平坦阶段
            }else {
                phase = 2; // 进入或保持递减阶段
            }
        }
    return true;
    }
}