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

题目

给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 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;
    }
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容