题目:题目链接
说来惭愧啊,这道题我真不知道咋写,只能想到暴力,但数据量10*5,O(n^2)铁定超时的。然后我悄悄地看了一眼discuss,看到一个单词“stack”,然后我就瞬间觉得有道理啊,我怎么没想到数据结构啊,干嘛一直从数学角度去想呢。然后我模拟了一下栈,插入数字,碰到大的就消掉,这不和消消乐差不多吗?我懂了!
我建议你可以先手动模拟一下,把数字一个个压到栈里面,然后每压入一个就和栈顶元素比较,比他大就消掉(也就是弹出栈),然后再重复上述步骤,直到栈为空或者小于栈顶元素。
我以样例一[73,74,75,71,69,72,76,73]为例来分析,初始栈S为空。
首先对73,S为空,啥也不干就压入73,S=[73]。
对74,74>73,记录下标,弹出73,压入74,S=[74]。
对75,75>74,记录下标,弹出74,压入75,S=[75]。
对71,71<75,压入71,S=[75,71]。
对69,69<71,压入69,S=[75,71,69]。
对72,弹出69,记录下标,弹出71,记录下标,S=[75,72]。
对76,弹出72,记录下标,弹出75,记录下标,栈为空,压入76,S=[76]。
对73,直接压入,S=[76,73]。
看文字挺麻烦的,建议你自己动手感受一下!
那么我们只需要解决怎么记录就可以了,要用到pair了,我们把数字和他的下标看成一个pair,然后记录就方便多了,因为我们知道了下标,就知道下标差,也就是答案。上代码:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int len = temperatures.size();
if (len == 1) return { 0 };
vector<int>ans(len, 0);
stack<pair<int, int>> s;/*将当前数字与下标都压入栈*/
for (int i = 0; i < len; ++i) {
while (!s.empty() && s.top().first < temperatures[i]) {
/*如果栈不为空.而且栈顶的值小于当前值就可以把它给“消掉”,
*然后记录此时压入数字栈的下标与消掉数字的下标差
*看代码就很清楚。
*/
ans[s.top().second] = i - s.top().second;
s.pop();
}
s.push({ temperatures[i],i });
}
return ans;
}
};
可能我表述得不是很清楚,我强烈建议你手动模拟一遍,根据上面的话也好,根据代码也好。因为我就是看到那个stack然后思考,最后手动模拟一遍的。我觉得你肯定也可以啊!
加油!