我的思路
尽管本题是在栈的目录下, 但是并没有想到什么栈的思路, 所以就先使用最直接的暴力法来解决 : 两个 for 循环, 找到比当前数值大的就立即跳出里层的循环 :
class Solution {
public int[] dailyTemperatures(int[] T) {
int length = T.length;
int[] ans = new int[length];
for (int i = 0; i < length; i++) {
int flag = 0;
for (int j = i; j < length; j++) {
if (T[j] > T[i]) {
ans[i] = j - i;
flag = 1;
break;
}
}
if (flag == 0) {
ans[i] = 0;
}
}
return ans;
}
}
这种方法的时间复杂度自然是很高的.
Runtime : 228 ms, faster than 7.63%
改进思路 : 考虑用一个哈希表来保存数值与位置, 进行排序就可以通过遍历来完成, 这个复杂度想了一下, 还是很高. 放弃.
栈的特点是先进后出, 如何与本题联系起来呢?
大神的思路
使用的是递减栈, 栈中元素对应的数组中的元素是依次递减的, 每一次 push 之前, 检查栈首元素对应的数组中的元素是否小于当前元素对应的数组中的元素, 如果是, 那么就找到了比它大的 temperature , 就是符合条件的, 求出坐标差值, 并出栈. 如果有多个都是小于, 就都出栈.
class Solution {
public int[] dailyTemperatures(int[] T) {
Stack<Integer> stack = new Stack<>();
int[] ret = new int[T.length];
for (int i = 0; i < T.length; i++) {
while(!stack.isEmpty() && T[i] > T[stack.peek()]) {
ret[stack.peek()] = i - stack.pop();
}
stack.push(i);
}
return ret;
}
}
以 T = [ 73, 74, 75, 71, 69, 72, 76, 73], 当出现 75, 71, 69 这是顺序递减的, 后面的 72 出现时, 就会在 while 循环中将前面比 72 小的 71, 69 都处理掉的.