题目链接
之前没接触过单调栈的题以为是思维题想了一个思路 时间超越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;
}