其他分享
首页 > 其他分享> > [亚麻高频题] Leetcode 503. Next Greater Element II(下一个大的元素)

[亚麻高频题] Leetcode 503. Next Greater Element II(下一个大的元素)

作者:互联网

题目描述 & 链接

Leetcode 503. 下一个更大元素

题目思路

1. 单调栈

单调栈经典应用 - 寻找下一个更大元素的位置,这道题不同的是数组是一个循环数组,也就是下一个更大元素位置也可以在当前元素前面。所以不仅要查看后面的元素,也要查看前面元素,对于循环数组的处理一般是将数组拼接成两份,这样循环数组的每一个值都会在当前元素位置中出现一次。然后算法思路根据 单调栈模板就可以实现。

代码如下:

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        
        // next greater element
        Deque<Integer> stk = new ArrayDeque<>();
        int[] res = new int[nums.length];
        Arrays.fill(res, -1);
        
        
        for(int i=0; i<nums.length*2; i++) {
            
            while(!stk.isEmpty() && nums[i%nums.length]>nums[stk.peek()]) {
                int idx = stk.pop();
                res[idx] = nums[i%nums.length];
            }
            
            stk.push(i%nums.length);
        }
        
        return res;
    }
}

时间复杂度:O(2N);空间复杂度:O(N)。 

标签:Greater,nums,int,res,元素,Element,数组,stk,Leetcode
来源: https://blog.csdn.net/ok1382038/article/details/122763304