代码随想录算法训练营第五十七天|739.每日温度、496.下一个更大元素I

739.每日温度
暴搜法，时间复杂度是O(n^2)

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        vector<int> result(temperatures.size(),0);
        for(int i=0;i<temperatures.size();i++)
        {
            bool flag=false;
            for(int j=i+1;j<temperatures.size();j++)
            {
                if(temperatures[j]>temperatures[i])
                {
                    flag=true;
                    result[i]=j-i;
                    break;
                }
            }
            if(!flag)
            {
                result[i]=0;
            }
        }
        return result;
    }
};

最后就出现超时！
单调栈：
使用栈这种数据结构来维持一个单调的序列！

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        vector<int> result(temperatures.size(),0);
        stack<int> stk;
        stk.push(0);
        for(int i=1;i<temperatures.size();i++)
        {
            if(temperatures[i]<=temperatures[stk.top()])
            {
                stk.push(i);
            }
            else
            {
                while(!stk.empty()&&temperatures[i]>temperatures[stk.top()])
                {
                    result[stk.top()]=i-stk.top();
                    stk.pop();
                }
                stk.push(i);
            }
        }
        return result;
    }
};

496.下一个更大元素I
思路：还是和上一道题的思路区别不大，但是关键还是转换到哈希表，综合性还是蛮大的！

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> result(nums1.size(),-1);
        stack<int>stk;
        if(nums1.size()==0)return result;
        unordered_map<int,int> mymap;
        for(int i=0;i<nums1.size();i++)
        {
            mymap[nums1[i]]=i;
        }
        stk.push(0);
        for(int i=1;i<nums2.size();i++)
        {
            if(nums2[i]<=nums2[stk.top()])
            {
                stk.push(i);
            }
            else
            {
                while(!stk.empty()&&nums2[i]>nums2[stk.top()])
                {
                    if(mymap.count(nums2[stk.top()])>0)
                    {
                        int index=mymap[nums2[stk.top()]];
                        result[index]=nums2[i];
                    }
                    stk.pop();
                }
                stk.push(i);
            }
        }
        return result;
    }
};