LeetCode Next Greater Element II

最新推荐文章于 2021-10-27 18:00:14 发布

IT独白者

最新推荐文章于 2021-10-27 18:00:14 发布

阅读量243

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode刷题

本文链接：https://blog.youkuaiyun.com/sun_wangdong/article/details/82177126

LeetCode刷题专栏收录该内容

201 篇文章

订阅专栏

博客围绕循环数组展开，要求输出每个元素的下一个更大元素，若不存在则输出 -1。题解指出该题与之前题目类似，但元素在循环数组中且可能多次出现。解决方法是生成新数组让原数组数字循环出现两遍，并用两个栈分别保存下标和元素值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a circular array (the next element of the last element is the first element of the array), print the Next Greater Number for every element. The Next Greater Number of a number x is the first greater number to its traversing-order next in the array, which means you could search circularly to find its next greater number. If it doesn't exist, output -1 for this number.

Example 1:

Input: [1,2,1]
Output: [2,-1,2]
Explanation: The first 1's next greater number is 2; 
The number 2 can't find next greater number; 
The second 1's next greater number needs to search circularly, which is also 2.

Note: The length of given array won't exceed 10000.

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

示例 1:

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

注意: 输入数组的长度不会超过 10000。

题解：这道题与之前的两道题有异曲同工之妙，都是寻找当前元素的第一个比它大的元素，不过之前的题目中，所有元素都是只出现一次，而这道题目中，元素是存在一个循环数组中，而且数字可能会出现多次。那么我就考虑再新生成一个数组，将原数组中的数字循环出现两遍，然后再用相同的方法来实现。不过这里，需要用两个栈来做，一个栈用来保存数组的下标，另一个保存数组中的元素值，保存下标的原因是因为该数组中的元素可能会多次出现，而不是出现1次。

public int[] nextGreaterElements(int[] nums)
    {
        int length = nums.length;
        int[] newnums = new int[2 * length];
        for(int i = 0; i < length; i++)
        {
            newnums[i] = nums[i];
            newnums[length + i] = nums[i];
        }
        Stack<Integer> stack1 = new Stack<>();            //存下标对应的元素
        Stack<Integer> stack2 = new Stack<>();            //存下标,因为该数组存在重复元素
        HashMap<Integer,Integer> map = new HashMap<>();   //存结果
        for(int i = 0; i < newnums.length; i++)
        {
            while(!stack1.isEmpty() && (stack1.peek() < newnums[i]))
            {                      //这里需要比较栈顶元素和当前元素的大小
                stack1.pop();
                map.put(stack2.pop(),newnums[i]);
            }
            stack1.push(newnums[i]);
            stack2.push(i);
        }
        int[] res = new int[length];
        for(int i = 0; i < length; i++)
            res[i] = -1;
        for(int i = 0; i < length; i++)
        {
            if(map.containsKey(i))
                res[i] = map.get(i);
        }
        return res;
    }

可以和之前的两篇博客对比，加深对这个问题的理解。