LeetCode739每日温度(思维或单调栈)

题目链接
在这里插入图片描述
之前没接触过单调栈的题以为是思维题想了一个思路 时间超越100%
倒着遍历 倒数第一个为0
依次遍历的时候先和第i+1的元素比较,如果第i+1个元素大于当前元素则答案数组赋值为1
如果第i+1个元素不比自己大,找比第i+1个元素大的值,直到找到第一个大于第i个元素的元素或者某个res[vis] = 0 则赋值为0 说明没有比当前元素大的值了。

    public int[] dailyTemperatures(int[] T) {
    
    
        int len = T.length;
        int res[] = new int[len];
        for (int i = len-2; i >= 0 ; i--) {
    
    
            if (T[i+1]>T[i]) res[i] = 1;//身后第一个温度的比自己大 赋1
            else {
    
    
                int vis = i+1;//找比身后第一个元素还大的元素
                while (res[vis] != 0){
    
    //如果没有则结束
                    vis += res[vis];
                    if (T[vis] > T[i]) {
    
    //如果有比自己大的元素 赋值退出循环
                        res[i] = vis - i;
                        break;
                    }
                }
            }
        }
        return res;
    }

第二个解法是单调栈
学习参考其中的视频
遍历整个数组,如果栈不空,且当前数字大于栈顶元素,那么如果直接入栈的话就不是 递减栈 ,所以需要取出栈顶元素,由于当前数字大于栈顶元素的数字,而且一定是第一个大于栈顶元素的数,直接求出下标差就是二者的距离。
继续看新的栈顶元素,直到当前数字小于等于栈顶元素停止,然后将数字入栈,这样就可以一直保持递减栈,且每个数字和第一个大于它的数的距离也可以算出来。

    public int[] dailyTemperatures(int[] T) {
    
    
        int len = T.length;
        int res[] = new int[len];
        Deque<Integer> stack = new LinkedList<>();
        for (int i = 0; i < len; i++) {
    
    
            while (!stack.isEmpty() && T[i] > T[stack.peek()]){
    
    //栈不为空并且大于栈顶元素更新答案
                int top = stack.pop();
                res[top] = i-top;
            }
            stack.push(i);
        }
        return res;
    }

猜你喜欢

转载自blog.csdn.net/qq_43434328/article/details/114668770