刷leetcode-day38
一、题目:
739.每日温度
根据每日 气温
列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 0
来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]
,你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]
。
二、思路:
遍历vector,对于每个元素,再遍历他之后的元素,如果存在比他大的元素,把他们之间的下标差放入一个新vector,否则将一个0放入新vector。
三、代码(C++):
vector<int> dailyTemperatures(vector<int>& T) {
vector<int> increase; //新建一个栈
int j=0;
for(int i=0;i!=T.size();i++) //遍历T中所有元素
{
for(j=i+1;j!=T.size();j++) //查找该元素后面是否存在比他大的数
{
if(T[j]>T[i])
{
increase.push_back(j-i); //如果有更大的数,把他们的下标差放入新的vector
break;
}
}
if(j==T.size())
increase.push_back(0); //不存在更大的数,把0放入新vector
}
return increase;
}
四、扩展:
暴力算法时间复杂度为O(n2),如果引入一个栈可以降低时间复杂度到O(n)。
思路:
遍历T,把下标放入一个栈,如果T[i]>T[s.top()],则新的vector[s.top()]=i-s.top();再把下标s.top[]出栈。最后生成一个新vector。
代码:
vector<int> dailyTemperatures(vector<int>& T) {
vector<int> increase(T.size()); //构造一个大小等于T的新vector,元素都为0
stack<int> s1; //构造存放下标的栈
for(int i=0;i!=T.size();i++)
{
while(!s1.empty()&&(T[i]>T[s1.top()])) //如果下标栈非空且T[i]>T[s.top()],则increase[s.top()]的值就为i-s1.top()
{
increase[s1.top()]=(i-s1.top());
s1.pop();
}
s1.push(i);
}
return increase;
}