根据每日 气温 列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
一。超时暴力
正常能想到的大暴力算法 , 双重for循环,第一层循环i从0到len , 第二层循环j从i+1到len , 通过判断存下索引的差值 , 但是35/37 超时辽。
二。暴力
逆序遍历列表存下每个值的索引 ( 注意这里并不是先一遍for循环存下每个值的索引 , 因为有重复的数字,会覆盖之前的值)
逆序的原因:最后一个数字一定是0 , 通过已知的数来确定前面的数 , 否则就要每次循环后面的数来确定有没有比当前值大。
温度的范围在[30, 100] , 那么如果当前值是T[i] , 只要找到(T[i]+1 ~100)之间与i最近的值就好了 ,因此用nex数组记录T[i]的索引 , 然后记下nex[T[i]+1] .......nex[100] 中最小的值与i的差值就是答案。
代码:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
int len = T.size();
int maxn = 1000000;
vector<int>ans(len);
int nex[200];
memset(nex , maxn , sizeof(nex));
for(int i = len-1 ; i >= 0 ; i--)
{
int op = maxn;
for(int j = T[i]+1 ; j <= 100 ; j++)
{
if(nex[j] < op)
{
op = nex[j];
}
}
if(op != maxn)
{
ans[i] = op-i;
}
nex[T[i]] = i;
}
return ans;
}
三。单调栈
单调栈中记录的是列表的下标 , 例如有1 4 2 7 3 4 , S为栈
- ------S(0) 栈中放入1的下标0
4 > 1 因为4>1 所以 0 出栈 ,4的下标1 入栈------1(1) 4 2 7 3 4
- ------S(1)
2 < 4 因为2<4 所以2的下标2 入栈
4--2-----S(1 , 2)
7 > 2 所以2的下标2出栈------1(1) 4 2(1) 7 3 4
又因为7>4 所以4的下标1出栈------1(1) 4(2) 2(1) 7 3 4
此时栈中S(7)
3 < 7 所以3的下标4入栈
S(7 , 4)
4>3 所以3的下标4出栈 ------1(1) 4(2) 2(1) 7 3(1) 4
4<7 所以4的下标5入栈 , 最后栈中剩的
S(7 , 5)均为0 ------1(1) 4(2) 2(1) 7(0) 3(1) 4(0)
代码:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
int len = T.size();
vector<int>ans(len);
stack<int>S;
for(int i = 0 ; i < len ; i++)
{
if(S.empty()) S.push(i);
else
{
while(!S.empty() && T[i] > T[S.top()])
{
ans[S.top()] = i - S.top();
S.pop();
}
S.push(i);
}
}
return ans;
}
};