LeetCode 845. 数组中的最长山脉

题目

我们把数组 A 中符合下列属性的任意连续子数组 B 称为 “山脉”：

• B.length >= 3
• 存在 0 < i < B.length - 1 使得 B[0] < B[1] < … B[i-1] < B[i] > B[i+1] > … > B[B.length - 1]
  （注意：B 可以是 A 的任意子数组，包括整个数组 A。）

给出一个整数数组 A，返回最长 “山脉” 的长度。

如果不含有 “山脉” 则返回 0。

---

思路

数组递增称为上坡，递减称为下坡，否则称为平坡。
用两个指针记录上坡起点（left）和下坡终点（right），两个状态记录上坡（up）和下坡(down)

• up状态下(A[i] > A[i-1])
  up = true; // 新山脉的上下坡状态更新，此时为上坡
  right = i; // 记录山脉右端点
  • 检查之前是否经历过down
    若经历过，说明此时遇到的是新山脉的起始。需要更新山脉状态和山脉起始位置。
    down = false; // 新山脉的上下坡状态更新，未经历下坡
    left = i-1; // 新山脉的起始点更新
• down状态下(A[i] < A[i-1])
  检查之前是否经历过up
  1. 经历过
     说明此时正经历山脉的下坡。需要更新山脉状态和山脉终点位置。
     down = true; // 新山脉的上下坡状态更新，经历下坡
     right = i; // 新山脉的起始点更新
  2. 未经历
     此时不符合山脉条件，是无效状态，初始化起点终点位置和上下坡状态
     left = i; right = i; up = false; down = false;
• 平坡状态下(A[i] = A[i-1])
  此时不符合山脉条件，是无效状态，初始化起点终点位置和上下坡状态
  left = i; right = i; up = false; down = false;
• 每相邻元素比较过一次，就要更新一下结果
  res是最长山脉，当前记录的山脉长度是right-left+1，确定经历过有效状态上下坡(if(up && down))条件后，两者取最大值更新。

---

以下是完整代码

class Solution {
public:
    int longestMountain(vector<int>& A) {
        int len = A.size();
        if(len < 3)
            return 0;
        int res = 0;
        int left = 0,right = 0;
        bool up = false, down = false;
        for(int i=1;i < len;i++){
            if(A[i] > A[i-1])
            {
                up = true;
                right = i;
                if(down){
                    left = i-1;                    
                    down = false;
                }     
            }
            else if(A[i] < A[i-1])
            {
                if(up){
                    right = i;
                    down = true;
                }
                else
                {
                    left = i;
                    right = i;
                    up = false; 
                    down = false;
                }
            }
            else
                {
                    left = i;
                    right = i;
                    up = false; 
                    down = false;
                }
            if(up&&down) res = max(res, right-left+1);
        }  
        return res;
    }
};