LeetCode 739.每日温度

利用单调栈保存数值降序的下标，遇到高温度则出栈，天数等于i-栈内值，遇到低温度则进栈

注意：可以连续出栈，因此内部需要设计一个连续出栈循环，注意边界条件为栈非空，且i<n

由于一些温度可能没有比其高的天，会始终被留在栈内，因此初始化时结果数组统一初始化为0，未处理的默认情况就是0

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <stack>
using namespace std;

/*
* 思路：利用单调栈，栈内存下标
* 遇到大数则出栈，遇到小数则进栈，由于存的是下标，不需要在意求解顺序
* 出栈时可以循环出栈（新入栈的数足够大时）
*/
class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        stack<int>s;
        int n = temperatures.size();
        vector<int>t(n, 0);
        s.push(0);
        for (int i = 1; i < n; i++)
        {
            int top = s.top();
            while (!s.empty() && i < n && temperatures[top] < temperatures[i])
            {
                t[top] = i - top;
                s.pop();
                if (!s.empty())
                    top = s.top();
            }
            s.push(i);
        }
        return t;
    }
};



int main()
{
    int n, a;
    cin >> n;
    vector<int>nums;
    for (int i = 0; i < n; i++)
    {
        cin >> a;
        nums.push_back(a);
    }
    Solution A;
    vector<int>temp = A.dailyTemperatures(nums);
    for (int i = 0; i < temp.size(); i++)
    {
        cout << temp[i] << " ";
    }
}