Leetcode 503.下一个更大元素

下一个更大元素

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

示例 1:

输入: [1,2,1]

输出: [2,-1,2]

解释: 第一个 1 的下一个更大的数是 2;

数字 2 找不到下一个更大的数;

第二个 1 的下一个最大的数需要循环搜索,结果也是 2。

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

思路

使用栈来进行优化上面的算法,我们遍历两倍的数组,然后还是坐标i对n取余,取出数字,如果此时栈不为空,且栈顶元素小于当前数字,说明当前数字就是栈顶元素的右边第一个较大数,那么建立二者的映射,并且去除当前栈顶元素,最后如果i小于n,则把i压入栈。因为res的长度必须是n,超过n的部分我们只是为了给之前栈中的数字找较大值,所以不能压入栈,参见代码如下:

 1 import java.util.Arrays;
 2 import java.util.Stack;
 3 
 4 class Solution {
 5     public int[] nextGreaterElements(int[] nums) {
 6         int n=nums.length;
 7         int[] res=new int[nums.length];
 8         Arrays.fill(res,-1);
 9         Stack<Integer> stack=new Stack<Integer>();
10         for(int i=0;i<2*n;i++){
11             int num=nums[i%n];
12             while(!stack.isEmpty()&&nums[stack.peek()]<num){
13                 res[stack.peek()]=num;
14                 stack.pop();
15             }
16             if(i<n) stack.push(i);
17         }
18         return res;
19     }
20 }

 

posted on 2019-02-14 00:20  kexinxin  阅读(89)  评论(0编辑  收藏  举报

导航