请根据每日气温列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],
你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
解题思路:
暴力搜索:
对下标 i 往后搜索,直到找到最小的 j ,满足 j>i,T[j]>T[i]
因为温度为[30,100]内的整数,维护一个[30,100]的数组Temperature记录出现的温度,初始值为无穷大,记录值为此温度出现的最早时间。
对T[i]从后往前搜索。查找T[i]+1的值在Temperature里出现的最小下标值minj,如果此值为无穷大,则赋值为0,否则赋值minj- i , Temperature[T[i]]=i.
因为对T从后往前更新,则保证在查找Temperature数组时,查找出的数一定是出现过的数。
单调栈:
维护一个栈,如果出现的数比栈内的数字大,则把栈内的数字取出来,为栈内的数字赋值,如果没有栈内的数大,则把此数也加进栈。
例 [73, 74, 75, 71, 69, 72, 76, 73]
第一步:
栈:[0(73)]
结果:[0,0,0,0,0,0,0,0]
第二步:74>73,取出0(73),ans[0]的结果是1-0=1
栈:[1(74)]
结果:[1,0,0,0,0,0,0,0]
第三步:75>74,取出1(73),ans[1]的结果是2-1=1
栈:[2(75)]
结果:[1,1,0,0,0,0,0,0]
第四步:71<75,存入3(71)
栈:[2(75),3(71)]
结果:[1,1,0,0,0,0,0,0]
第五步:69<71,存入4(69)
栈:[2(75),3(71),4(69)]
结果:[1,1,0,0,0,0,0,0]
第六步:72>69,取出由于 72大于 69 和 71,因此依次移除栈顶元素 4 和 3,赋值 ans[4]:=5-4和ans[3]:=5−3,将 5 进栈。
栈:[2(75),5(72)]
结果:[1,1,0,2,1,0,0,0]
第七步:由于 76大于 72 和 75,因此依次移除栈顶元素 5 和 2,赋值 ans[5]:=6-5和 ans[2]:=6-2,将 6 进栈。
栈:[6(76)]
结果:[1,1,4,2,1,1,0,0]
第八步:由于 73 小于 76,因此将 7 进栈。
栈:[6(76),7(73)]
结果:[1,1,4,2,1,1,0,0]
public int[] dailyTemperatures(int[] T) {
int length = T.length;
int[] ans = new int[length];
Stack<Integer> s = new Stack<Integer>();
for(int i=0;i<length;i++){
while(!s.isEmpty()&&T[i]>T[s.peek()]){
int m = s.peek();
s.pop();
ans[m] = i-m;
}
s.push(i);
}
return ans;
}