leetcode 896.单调数列

leetcode 896.单调数列

题干

如果数组是单调递增或单调递减的，那么它是单调的。
如果对于所有 i <= j，A[i] <= A[j]，那么数组 A 是单调递增的。 如果对于所有 i <= j，A[i]> = A[j]，那么数组 A 是单调递减的。
当给定的数组 A 是单调数组时返回 true，否则返回 false。

示例 1：
输入：[1,2,2,3]
输出：true

示例 2：
输入：[6,5,4,4]
输出：true

示例 3：
输入：[1,3,2]
输出：false

示例 4：
输入：[1,2,4,5]
输出：true

示例 5：
输入：[1,1,1]
输出：true

提示：
1 <= A.length <= 50000
-100000 <= A[i] <= 100000

题解

模拟，这个写法要注意开头数一样大的情况

class Solution {
public:
    bool isMonotonic(vector<int>& A) {
        int n = A.size();
        if(n == 0 || n == 1) return true;
        int flag = A[1] - A[0] == 0 ? 0 : A[1] - A[0] > 0 ? 1 : -1;
        for(int i = 2 ; i < n ; ++i){
            if(flag == 0 && A[i] != A[i-1]){
                flag = A[i] - A[i - 1] > 0 ? 1 : -1;
            }
            cout<<A[i] - A[i-1]<<' '<<flag<<endl;
            if((A[i] - A[i - 1]) * flag < 0) return false;
        }
        return true;
    }
};

改写一下，只用严格遍历A一次

class Solution {
public:
    bool isMonotonic(vector<int>& A) {
        int n = A.size();
        if(n == 0 || n == 1) return true;
        int flag = 0;
        int head = 0;
        for(int i = 1 ; i < n ; ++i){
            if(A[i] != A[i - 1]){
                flag = A[i] - A[i - 1];
                head = i + 1;
                break;
            }
        }
        if(head == 0){
            return true;
        }
        for(int i = head ; i < n ; ++i){
            if((A[i] - A[i - 1]) * flag < 0) return false;
        }
        return true;
    }
};

或者再改写一下，两个变量标记增减情况是否出现过

class Solution {
public:
    bool isMonotonic(vector<int>& A) {
        int n = A.size();
        if(n == 0 || n == 1) return true;
        bool flagUp = false,flagDown = false;
        for(int i = 1 ; i < n ; ++i){
            if(!flagUp && A[i] - A[i - 1] > 0) flagUp = true;
            if(!flagDown && A[i] - A[i - 1] < 0) flagDown = true;
        }
        return (!flagDown) || (!flagUp);
    }
};