力扣解题思路:503. 下一个更大元素 II 纠错记录

503. 下一个更大元素 II


思路:由题目,给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
这一题我主要来记录我的错误原因,首先如果这一题的题目不是循环数组,那么这一题用栈来解决很简单,看代码:

public int[] nextGreaterElements(int[] nums) {
    int[] res = new int[nums.length];
    Arrays.fill(res,-1);
    int n = nums.length;
    //System.out.println(nums.length-1);
    Stack<int[]> stack = new Stack<>();
    for(int i=0;i<n;i++){
        while(!stack.isEmpty()&&nums[i]>stack.peek()[0]){
            //System.out.println(i);
            res[stack.peek()[1]] = nums[i];
            stack.pop();
        }
        stack.push(new int[]{nums[i],i});
    }
    return res;
}

考虑到循环数组的存在,那么我们可以将数组拼接成两个,这样头和尾自然就联结在一起了:

public int[] nextGreaterElements(int[] nums) {
    int[] res = new int[nums.length];
    Arrays.fill(res,-1);
    int n = nums.length;
    //System.out.println(nums.length-1);
    Stack<int[]> stack = new Stack<>();
    for(int i=0;i<n*2;i++){
        i = i%nums.length;
        //System.out.println(i);
        while(!stack.isEmpty()&&nums[i]>stack.peek()[0]){
            //System.out.println(i);
            res[stack.peek()[1]] = nums[i];
            stack.pop();
        }
        stack.push(new int[]{nums[i],i});
    }
    return res;
}

然后我就这样写,就超出时间限制了/(ㄒoㄒ)/~~
开始找原因,首先超时一定就是没能出循环(因为超时的测试用例很短),先找while,很明显不是while,因为while中由pop操作,stack一定会有为空的时候,所以就是for循环的问题了,果然i有问题,每次i都被我重新赋值,永远都走不到终点自然就超时了,所以不能直接对i赋值,修改如下:

public int[] nextGreaterElements(int[] nums) {
    int[] res = new int[nums.length];
    Arrays.fill(res,-1);
    int n = nums.length;
    //System.out.println(nums.length-1);
    Stack<int[]> stack = new Stack<>();
    for(int i=0;i<n*2;i++){
        //i = i%nums.length;
        //System.out.println(i);
        while(!stack.isEmpty()&&nums[i%nums.length]>stack.peek()[0]){
            //System.out.println(i);
            res[stack.peek()[1]] = nums[i%nums.length];
            stack.pop();
        }
        stack.push(new int[]{nums[i%nums.length],i%nums.length});
    }
    return res;
}

这样就没问题啦,当然stack中可以不存放数组,这只是我的个人习惯,也可以这样写,也是一样的啦o( ̄▽ ̄)ブ

public int[] nextGreaterElements(int[] nums) {
    int n = nums.length;
    int[] next = new int[n];
    Arrays.fill(next, -1);
    Stack<Integer> pre = new Stack<>();
    for (int i = 0; i < n * 2; i++) {
        int num = nums[i % n];
        while (!pre.isEmpty() && nums[pre.peek()] < num) {
            next[pre.pop()] = num;
        }
        if (i < n){
            pre.push(i);
        }
    }
    return next;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值