【打基础】算法集

单调栈的应用

2019-10-10  本文已影响0人  拜仁的月饼

1. 例题引入

来源LeetCode739-每日温度

原题

根据每日气温列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。

例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/daily-temperatures
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& T) {
        vector<int> res(T.size(), 0); // 返回结果时候用。向量的大小与输入的气温列表T相同
        stack<int> tmp; // 维护单调栈
        for(int i = 0; i < T.size(); i++)
        {
            while(!tmp.empty() && T[tmp.top()] < T[i])
            {
                res[tmp.top()] = (i - tmp.top());
                tmp.pop();
            }
            tmp.push(i);
        }
        return res;
    }
};
上一篇 下一篇

猜你喜欢

热点阅读