Leetcode刷题笔记题解(C++):739. 每日温度

Leetcode刷题笔记题解(C++):739. 每日温度_第1张图片

### 解题思路:利用栈来解决。

注意的是栈中压入的下标的值

首先压入0

如果栈不为空且要压入的下标值i有T[i]>T[栈顶元素] 则res[栈顶元素]=i-栈顶元素

否则继续压栈,直到压完T中所有的数据为止。

还要注意的是,开始对res全部赋值为0,这就对栈中的元素没弹出来的默认为0;

这里附一下图吧, 好理解一点。

Leetcode刷题笔记题解(C++):739. 每日温度_第2张图片

代码如下:

 

class Solution {
public:
    vector dailyTemperatures(vector& T) {
        int n = T.size();
        //初始化vector数据,初始全部为0;
        vector res(n, 0);
        stack st;
        for (int i = 0; i < n; ++i) {
            //如果栈不为空,且选取的值大于下标为栈顶的值对应的数组值,则天数为下标之间的差
            while (!st.empty() && T[i] > T[st.top()]) {
                res[st.top()] = i - st.top();
                st.pop();
            }
            //栈为空或者栈顶元素下标对应的值大于要压如栈的值
            st.push(i);
        }
        return res;
    }
};

 

你可能感兴趣的:(Leetcode算法题解,c++,栈算法,leetcode)