栈---下一个更大元素

本文介绍了如何使用栈解决在一个循环数组中查找每个元素的下一个更大元素的问题。通过维护栈顶元素递增的序列,当遍历数组时,找到比栈中元素更大的数并更新答案,若找不到则标记为-1。
  1. 题目
    给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
  2. 示例
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
  1. 代码实现
    数据结构:栈;维护栈顶元素最小的递增的序列。
    a.常规算法找下一个最大元素;
    b.nums数组最后一个元素入栈,遍历nums数组,找比栈中元素下一个更大的数;
    c.未找到下一个更大的元素,置为-1.
import java.util.Stack;

public class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int[] array = new int[nums.length];
        Stack<StackElement> stack = new Stack<>();
        //常规找下一个更大的元素
        for (int i=0;i<nums.length-1;i++){
            StackElement stackElement = new StackElement();
            stackElement.val = nums[i];
            stackElement.index = i;
            stack.push(stackElement);
            int next = nums[i+1];
            while (!stack.empty()&&stack.peek().val<next){
                stackElement = stack.pop();
                array[stackElement.index] = next;
            }
        }
        //nums数组最后一个元素入栈,注:nums为空的情况
       if(nums.length>0){
           StackElement stackElement = new StackElement();
           stackElement.val = nums[nums.length-1];
           stackElement.index = nums.length-1;
           stack.push(stackElement);
       }
       //找栈中元素的下一个更大的数
        for(int i=0;i<nums.length&&!stack.empty();i++){
            while (!stack.empty()&&stack.peek().val<nums[i]){
                StackElement stackElement = stack.pop();
                array[stackElement.index] = nums[i];
            }
        }
        //未找到
        while (!stack.empty()){
            StackElement stackElement = stack.pop();
            array[stackElement.index] = -1;
        }
        return array;
    }
}

class StackElement{
    int val;
    int index;
}

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值