739. 每日温度/C++

最新推荐文章于 2024-08-04 22:12:12 发布

原创最新推荐文章于 2024-08-04 22:12:12 发布 · 806 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode/C++ 专栏收录该内容

248 篇文章

订阅专栏

本文介绍了一种使用单调栈解决“每日温度”问题的方法，通过遍历温度数组，找到每个元素右侧第一个比其大的元素，从而计算出等待天数。文章详细解释了算法原理，并提供了简化版代码实现。

在这里插入图片描述
使用单调栈，可以找到一个元素右边第一个比该元素大或者小的元素

vector<int> dailyTemperatures(vector<int>& T) {
   if(T.empty()) return vector<int>();
   int size=T.size();
   T.push_back(101);//在最后面压入一个最大值，以使得所有元素都可以出栈

   vector<int> res(size);
   stack<int> stk;
   for(int i=0;i<size+1;++i){
       while(!stk.empty() && T[i]>T[stk.top()]){
           if(i==size) //对于我们手动加入的元素要特殊处理
               res[i]==0;
           else{
               int top=stk.top();
               res[top]=i-top;
           }
           stk.pop();
       }
       stk.push(i);
   }
   return res;
}

但是这里有一个巧合，在没有添加101的前提下，如果一个元素没有出栈，就说明在它右边没有比它大的元素，也就是置为0，而数组的初始值正好为0，因此可以略微简化为

vector<int> dailyTemperatures(vector<int>& T) {
	if(T.empty()) return vector<int>();
   	int size=T.size();

   vector<int> res(size);
   stack<int> stk;
   for(int i=0;i<size;++i){
       while(!stk.empty() && T[i]>T[stk.top()]){
           int top=stk.top();
           res[top]=i-top;
           stk.pop();
       }
       stk.push(i);
   }
   return res;
}