并不难的一道题,主要思路就是遍历时将所有日期入栈,同时比较当前日期的温度是不是比栈里的日期所对应的温度高,如果是的话就计算出ret[]的结果。
需要注意的是栈中的元素永远是从小到大(栈顶到栈底)的顺序排列的(如果栈顶的温度比当前温度底的话就出栈了),所有这个算法才成立。
class Solution {
public int[] dailyTemperatures(int[] T) {
int[] ret = new int[T.length];
Stack<Integer> stack = new Stack<Integer>();
for(int i=0; i < T.length; i++){
while(!stack.isEmpty() && T[i] > T[stack.peek()]){
int index = stack.pop();
ret[index] = i - index;
}
stack.push(i);
}
return ret;
}
}