题目
思路
暴力
两重循环
递减栈
遍历整个数组,如果栈不空,且当前数字大于栈顶元素,那么如果直接入栈的话就不是递减栈,所以需要取出栈顶元素,由于当前数字大于栈顶的数字,而且一定是第一个大于栈顶元素的数,直接求出下标差就是二者的距离
继续看新的栈顶元素,直到当前数字小于等于栈顶元素停止,然后将数字入栈,这样就可以一直保持递减栈,且每个数字和第一个大于它的数的距离也可以算出来
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int length = temperatures.length;
int[] ans = new int[length];
// 原来deque的实现类也可以是LinkedList
// 栈内存储温度数组下标
Deque<Integer> stack = new LinkedList<Integer>();
for (int i = 0; i < length; i++) {
int temperature = temperatures[i];
while (!stack.isEmpty() && temperature > temperatures[stack.peek()]) {
int prevIndex = stack.pop();
ans[prevIndex] = i - prevIndex;
}
stack.push(i);
}
return ans;
}
}