根据每日 气温 列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 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;
}
}