LeetCode 845. 数组中的最长山脉

Posted 2021-09-01 数据结构和算法

截止到目前我已经写了 500多道算法题，其中部分已经整理成了pdf文档，目前总共有1000多页（并且还会不断的增加），大家可以免费下载
下载链接：https://pan.baidu.com/s/1hjwK0ZeRxYGB8lIkbKuQgQ
提取码：6666

解法一

山脉数组就是数组的前部分都是上升的，剩下的部分都是下降的，这样的数组称为山脉数组，之前专门讲过山脉数组，具体可以看下475，有效的山脉数组。

这里我们先找到上升的数组元素个数，到最高点之后再找下降的元素个数，他们相加再加上最顶端的元素个数（1）就是山脉数组的长度，我们只需要保留最长的即可。这里就以示例一为例来画个图看一下

来看下代码

public int longestMountain(int[] A) {
    int length = A.length;
    int max = 0;//保存最长的山脉长度
    int index = 1;
    while (index < length) {
        int up = 0;//上升的个数
        int down = 0;//下降的个数

        //既不上升也不下降的要过滤掉
        while (index < length && A[index - 1] == A[index])
            index++;
        //统计上升的个数
        while (index < length && A[index - 1] < A[index]) {
            index++;
            up++;
        }
        //统计下降的个数
        while (index < length && A[index - 1] > A[index]) {
            index++;
            down++;
        }
        //上升和下降的个数必须都大于0，才能称为山脉，计算山脉的长度，
        //保留最大的即可
        if (up > 0 && down > 0)
            max = Math.max(max, up + down + 1);
    }
    return max;
}

其实我们还可以提前计算好每一个元素左边上升的个数和右边下降的个数，如果某个元素左边是上升的并且右边是下降的，那么这个元素就是山脉数组中最大的元素，也就是山顶，我们需要计算这个山脉数组的长度。如下图所示

来看下代码。

public int longestMountain(int[] A) {
    int length = A.length;
    //up表示上升元素的个数
    int[] up = new int[length];
    for (int i = 1; i < length; i++) {
        if (A[i] > A[i - 1])
            up[i] = up[i - 1] + 1;
    }
    //down表示下降元素的个数
    int[] down = new int[length];
    for (int i = length - 1; i > 0; i--) {
        if (A[i - 1] > A[i])
            down[i - 1] = down[i] + 1;
    }
    //保留最大的长度
    int max = 0;
    for (int i = 0; i < length; i++) {
        if (up[i] == 0 || down[i] == 0)
            continue;
        max = Math.max(max, up[i] + down[i] + 1);
    }
    return max;
}

或者还可以合并一起，这样代码就越来越简洁了。

public int longestMountain(int[] A) {
    int max = 0, up = 0, down = 0;
    for (int i = 1; i < A.length; ++i) {
        //到山脚的拐点了，或者既没有上升也没有下降，把up和down
        //重新赋值
        if (down > 0 && A[i - 1] < A[i] || A[i - 1] == A[i])
            up = down = 0;
        //计算上升的长度
        if (A[i - 1] < A[i])
            up++;
        //计算下降的长度
        if (A[i - 1] > A[i])
            down++;
        //既有上升又有下降，说明是山脉数组，他是他的长度，
        //保留最长的即可
        if (up > 0 && down > 0)
            max = Math.max(max, up + down + 1);
    }
    return max;
}