美文网首页算法练习
下一个更大元素 II(LeetCode 503)

下一个更大元素 II(LeetCode 503)

作者: 倚剑赏雪 | 来源:发表于2020-03-07 23:03 被阅读0次

    题目

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

    示例 1:

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

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

    题解

    题解一:

    • 使用双重循环
    • 第二层循环从j = 1开始,每次取(i+j)%nums[nums.Length]
    • 时间复杂度为O(n²),空间复杂度为O(n)
    • 代码
     public int[] NextGreaterElements(int[] nums) {
                int len = nums.Length;
                int[] res = new int[len];
                for (int i = 0; i < len; i++)
                {
                    res[i] = -1;
                    for (int j = 1; j < len; j++)
                    {
                        if (nums[(j+i) % len] > nums[i])
                        {
                            res[i] = nums[(j+i) % len];
                            Console.WriteLine(res[i]);
                            break;
                        } 
                    }
                }
    
                return res;
        }
    

    题解二:

    • 使用单调栈
    • 假设存在一个[1,2,1,1,2,1]改造数组,这样一次循环的数组
    • 维持一个Stack记录元素的index,Stack中的index对应的元素单调递减,
    • 从改造数组的最后一个元素向前扫描
    • while当前元素比Stack的顶部元素大or Stack为空:
      Stack.Pop
      Stack为空?当前元素的结果是-1:Stack.Peek
    • 时间复杂度O(n),空间复杂度O(n)
    • 代码
        public int[] NextGreaterElements(int[] nums) {
      int len = nums.Length;
                int[] res = new int[len];
                Stack<int> stack = new Stack<int>();
                for (int i = len*2-1; i >= 0; --i)
                {
                    while (stack.Count>0&& nums[stack.Peek()]<=nums[i%len])
                    {
                        stack.Pop();
                    }
                    res[i % len] = stack.Count > 0 ? nums[stack.Peek()] : -1;
                    stack.Push(i % len );
                }
                return res;
        }
    

    相关文章

      网友评论

        本文标题:下一个更大元素 II(LeetCode 503)

        本文链接:https://www.haomeiwen.com/subject/oighdhtx.html