代码随想录60期day51

最新推荐文章于 2025-11-24 15:27:16 发布

原创最新推荐文章于 2025-11-24 15:27:16 发布 · 164 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode #职场和发展

单调栈

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& T) {
        // 递增栈
        stack<int> st;
        vector<int> result(T.size(), 0);
        st.push(0);
        for (int i = 1; i < T.size(); i++) {
            if (T[i] < T[st.top()]) {                       // 情况一
                st.push(i);
            } else if (T[i] == T[st.top()]) {               // 情况二
                st.push(i);
            } else {
                while (!st.empty() && T[i] > T[st.top()]) { // 情况三
                    result[st.top()] = i - st.top();
                    st.pop();
                }
                st.push(i);
            }
        }
        return result;
    }
};

下一个更大的元素

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        
        stack<int>st;
        vector<int>result(nums1.size() , - 1);
        if(nums1.size() == 0) return result;

        unordered_map<int,int>unmap; // key is the value that index points to  whle value is the index
        for(int i = 0 ; i<nums1.size() ;i++){
            unmap[nums1[i]] = i;
        }
        st.push(0);

        for(int i = 1 ; i < nums2.size() ; i ++){
            // case 1
            if(nums2[i] < nums2[st.top()]){
                st.push(i);
            }else if(nums2[i] == nums2[st.top()]){
                st.push(i);
            }else{
                while(!st.empty() && nums2[i] >nums2[st.top()]){
                    if(unmap.count(nums2[st.top()]) > 0){
                        int index = unmap[nums2[st.top()]];
                        result[index] =  nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }

        }
        return result;
    }
};

503

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> result(nums.size(), -1);
        if (nums.size() == 0) return result;
        stack<int> st;
        st.push(0);
        for (int i = 1; i < nums.size() * 2; i++) { 
            // 模拟遍历两边nums，注意一下都是用i % nums.size()来操作
            if (nums[i % nums.size()] < nums[st.top()]) st.push(i % nums.size());
            else if (nums[i % nums.size()] == nums[st.top()]) st.push(i % nums.size()); 
            else {
                while (!st.empty() && nums[i % nums.size()] > nums[st.top()]) {
                    result[st.top()] = nums[i % nums.size()];
                    st.pop();
                }
                st.push(i % nums.size());
            }
        }
        return result;
    }
};