LeetCode_单调栈_中等_503.下一个更大元素 II

布满荆棘的人生 2023-10-07 20:06 50阅读 0赞

#### 目录 ####

*  1.题目
 *  2.思路
 *  3.代码实现（Java）

## 1.题目 ##

给定一个循环数组 nums （nums\[nums.length - 1\] 的下一个元素是 nums\[0\]），返回 nums 中每个元素的下一个更大元素。

数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。

**示例 1：**  
输入: nums = \[1,2,1\]  
输出: \[2,-1,2\]  
解释: 第一个 1 的下一个更大的数是 2；  
数字 2 找不到下一个更大的数；  
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。

**示例 2：**  
输入: nums = \[1,2,3,4,3\]  
输出: \[2,3,4,-1,4\]

**提示：**  
1 <= nums.length <= 104  
\-109 <= nums\[i\] <= 109

来源：力扣（LeetCode）  
链接：https://leetcode.cn/problems/next-greater-element-ii

## 2.思路 ##

（1）循环遍历  
① 定义长度为 nums.length 的数组 res 用于保存最终结果；  
② 遍历数组 nums，对于当前元素 nums\[i\]，循环遍历数组 nums\[i + 1…nums.length - 1\]、nums\[0…i - 1\]，查找下一个更大元素，在代码实现方面，可以通过**下标取余**的方式来实现循环遍历，如果找不到，则令 res\[i\] = -1；  
③ 遍历结束后，直接返回 res 即可。

（2）单调栈  
思路参考[该 LeetCode 用户题解][LeetCode]。

## 3.代码实现（Java） ##

//思路1————循环遍历
    class Solution {
        
        public int[] nextGreaterElements(int[] nums) {
        
            int length = nums.length;
            int[] res = new int[length];
            for (int i = 0; i < length; i++) {
        
            	// flag = false 表示当前元素 nums[i] 不存在下一个更大元素
                boolean flag = false;
                for (int j = i + 1; j < i + length; j++) {
        
                    int num = nums[j % length];
                    if (num > nums[i]) {
        
                        res[i] = num;
                        flag = true;
                        break;
                    }
                }
                if (!flag) {
        
                    res[i] = -1;
                }
            }
            return res;
        }
    }

//思路2————单调栈
    class Solution {
        
        public int[] nextGreaterElements(int[] nums) {
        
            int length = nums.length;
            // res 用于存储最终结果
            int[] res = new int[length];
            Arrays.fill(res, -1);
            Stack<Integer> stack = new Stack<>();
            for (int i = 0; i < 2 * length; i++) {
        
                while (!stack.isEmpty() && nums[i % length] > nums[stack.peek()]) {
        
                    //栈不为空且当前元素比栈顶元素大，则说明当前元素是前面一些元素的下一个更大元素
                    res[stack.pop()] = nums[i % length];
                }
                stack.push(i % length);
            }
            return res;
        }
    }

[LeetCode]: https://leetcode.cn/problems/next-greater-element-ii/solution/dong-hua-jiang-jie-dan-diao-zhan-by-fuxu-4z2g/