LeetCode 503-Next Greater Element II（单调栈，下一个最大/小元素）

最新推荐文章于 2024-10-06 13:05:10 发布

祝如愿~

最新推荐文章于 2024-10-06 13:05:10 发布

阅读量44

点赞数

分类专栏： # 单调栈文章标签： leetcode 算法数据结构 c++

本文链接：https://blog.csdn.net/PriscillaZhao/article/details/128768162

版权

单调栈专栏收录该内容

6 篇文章 0 订阅

订阅专栏

LeetCode 503 Next Greater Element II

题干：

给定一个环形数组 $n u m s$ （ $n u m s [n u m s . l e n g t h - 1]$ 就是 $n u m s [0]$ ），对于其中每一个元素，求它之后比它大的第一个元素是多少。

Input: nums = [2,1,2,4,3]
Output: [4,2,4,-1,4]

解：

还是求下一个更大元素，不同的是数组是环形。

环形数组，常用套路是把数组长度翻倍，把原数组往后复制一份。但其实可以不用额外创建空间，只是把遍历的下标 $i$ 翻倍了，利用取模运算（ $\% n$ ）就可完成循环遍历。

vector<int> nextGreaterElements(vector<int>& nums) {
        int n = nums.size();
        stack<int> st;
        vector<int> ans(n); //给vector初始化，以便可以直接通过下标访问 
        for(int i = 2*n - 1; i >= 0; i--){
            while(!st.empty() && st.top() <= nums[i%n]){
                st.pop();
            }
            ans[i%n] = st.empty()? -1: st.top();//ans[i%n]覆盖更新，刷新了一遍，保证最后一次是准确答案
            st.push(nums[i%n]);
        }
        return ans;
    }

Tips：

循环遍历数组：取模（ $\% n$ ）
vector不能用下标形式添加元素，但是可以用下标形式对元素内容进行修改，即下标 $v [n]$ 是返回 $v$ 中第 $n$ 个位置元素的引用。
```
vector<int> v;
v[0]=1; //错误，因为v不包含任何元素
vector<int> v2(10);
v2[0]=1; //正确，此时v2是含有10个0的容器
```