LeetCode 503: 下一个更大元素 II

LeetCode 503: 下一个更大元素 II

解答

单调栈，但中间有一些小细节需要注意

代码一

这里需要特别注意的是单调栈中存的是值还是下标索引，如果存值的话（有时候写起来会简单一些），需要搭配map的(val, idx)映射，适合用于数组中没有重复值的情况；如果存下标索引的话，感觉这个更通用，但是有的题存值更方便一些
这个题目因为数组中有重复值，数组中重复值的nxtGreaterElem很可能不同，不能用存值的方法。
还有就是拿一个比较典型的例子来初步验证想法，比如一开始想的是两个数组拼起来+map+单调栈，但是用例子试一试就知道不行了（自己思维有点慢）

// my solution 
class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> ans(nums.size(), -1);
        stack<int> st;// format: idx
        int now=0;
        for(int i=0; i<nums.size(); i++){
            now = nums[i];
            if(st.empty()) st.push(i);
            else if( nums[st.top()] >= now) st.push(i);
            else{ // now>nums[st.top()]
                while(!st.empty() && now>nums[st.top()]){
                    ans[ st.top() ] = now;
                    st.pop();
                }
                st.push(i);
            }
        }

        for(int i=0; i<nums.size(); i++){
            now = nums[i];
            if(st.empty()) st.push(i);
            else if( nums[st.top()] >= now) st.push(i);
            else{ // now>nums[st.top()]
                while(!st.empty() && now>nums[st.top()]){
                    ans[ st.top() ] = now;
                    st.pop();
                }
                st.push(i);
            }
        }

        return ans;

    }
};

代码二

首先学习一下如何不写循环将两个数组拼起来

// 版本一
class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        // 用一个数组初始化另一个数组，其他一些容器也有类似的begin end初始化方法
        vector<int> nums1(nums.begin(), nums.end());
        // insert
        nums.insert(nums.end(), nums1.begin(), nums1.end());
        // 用新的nums大小来初始化result
        vector<int> result(nums.size(), -1);
        if (nums.size() == 0) return result;

        // 开始单调栈
        stack<int> st;
        for (int i = 0; i < nums.size(); i++) {
            while (!st.empty() && nums[i] > nums[st.top()]) {
                result[st.top()] = nums[i];
                st.pop();
            }
            st.push(i);
        }
        // 最后再把结果集即result数组resize到原数组大小
        result.resize(nums.size() / 2);
        return result;
    }
};

当然因为循环数组，没必要数组扩容

// 版本二
class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> result(nums.size(), -1);
        if (nums.size() == 0) return result;
        stack<int> st;
        for (int i = 0; i < nums.size() * 2; i++) {
            // 模拟遍历两边nums，注意一下都是用i % nums.size()来操作
            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;
    }
};