下一个较大元素

  • 现在我们有一个int数组,请你找出数组中每个元素的下一个比它大的元素。给定一个int数组A及数组的大小n,请返回一个int数组,代表每个元素比他大的下一个元素,若不存在则为-1。保证数组中元素均为正整数。
  • 测试样例:[11,13,10,5,12,21,3],7 返回:[13,21,12,12,21,-1,-1]
/*从后向前维护一个递减栈。
最右边的那个值肯定没有最大值,所以肯定是-1。初始栈为-1。
从后向前计算:
(1)如果当前元素大于栈顶元素,则栈顶元素退出,如果还是大于栈顶元素,继续退出,一直遍历栈到-1或者小于栈顶元素。这个元素就是就是当前值的下一个比较大的元素。
(2)如果当前元素小于栈顶元素,栈顶元素就是当前值的下一个比较大的元素。
再简化一下代码如下
*/

class NextElement {
public:
    vector<int> findNext(vector<int> A, int n){
        stack<int> s;
        s.push(-1);
        int sz = A.size();
        vector<int> res(sz);
        int top;
        for(int i = sz - 1; i >= 0; --i){
            top = s.top();
            while(top != -1 && A[i] >= top){
                s.pop();
                top = s.top();
            }
            s.push(A[i]);
            res[i] = top;
        }
        return res;
    }
};
  • 这题还有一个变种是这样的:
  • 现在有一个数组,请找出数组中每个元素的后面比它大的最小的元素,若不存在则为-1。
  • 也就是记录的时候丢弃的数组要将其重新push到栈中来,若只是重复上题的做法的话可能会导致找出的数不是最小的。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值