单调栈

最新推荐文章于 2024-10-28 16:12:02 发布

canaryW

最新推荐文章于 2024-10-28 16:12:02 发布

阅读量290

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/cobracanary/article/details/90576785

leetcode 专栏收录该内容

44 篇文章

订阅专栏

博客围绕根据每日气温列表生成对应等待升温天数列表的问题展开。给出示例及气温列表范围和值的范围，指出暴力解法时间复杂度为n^2，而运用单调栈可将时间复杂度降为O(n)，还介绍了维护栈和遍历数组的具体操作。

根据每日气温列表，请重新生成一个列表，对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高，请输入 0 来代替。

例如，给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示：气温列表长度的范围是 [1, 30000]。每个气温的值的都是 [30, 100] 范围内的整数。

其实这个就是要找出比当前数字大的最近的右边的下标，如果用暴力的话会是n^2，但是运用一个单调栈的话就可以把时间复杂度变成O(n)；

我维护一个栈，用一个index从左向右遍历数组，当当前的数字比栈顶元素小的时候直接入栈，否则一直pop直到栈空或者栈顶元素比当前的元素大，将当前元素入栈。

class Solution {
    public int[] dailyTemperatures(int[] T) {
        int len = T.length;
        int index = 0;
        Stack<Integer> stack = new Stack<Integer>();
        int[] ans = new int[len];
        while(index < len) {
        	if(stack.empty() || T[index] <= T[stack.peek()]) {
        		stack.push(index);
        	} else {
        		while(!stack.empty() && T[index] > T[stack.peek()]) {
        			int temp = stack.pop();
        			ans[temp] = index - temp;
        		}
        		stack.push(index);
        	}
        	index++;
        }
        return ans;
    }
}