LeetCode739

题目

给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指在第 i 天之后，才会有更高的温度。如果气温在这之后都不会升高，请在该位置用 0 来代替。

方法1：双重循环。缺点：效率太低

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {

        int num = temperatures.size();//根据数组temperatures的长度来创建数组。
        vector<int> answer(num);//创建answer数组
        for(int j=0;j<num;j++)
        {
            int after = 0;//记录几天后温度会升高
            for(int i=j+1;i<num;i++)
            {
                if(temperatures[i]>temperatures[j])
                {
                    answer[j] = after+1;
                    break;
                }
                else
                {
                after = after + 1;   
                }
            }
        }
        answer[num-1] = 0;
        return  answer;
    }
};

方法2：单调栈

• 单调栈中存放的是元素的下标

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {

        int num = temperatures.size();//数组的长度
        vector<int> answer(num); //创建结果数组
        stack<int> st;//创建一个堆
        st.push(0);
        for(int i = 1; i < num;i++)
        {
            while(!st.empty() && temperatures[i] > temperatures[st.top()])//当下一个元素大于栈顶元素
            {
                answer[st.top()] = i - st.top();
                st.pop();
            }
            st.push(i);
            }

       return answer;
    }
};