下一个较大元素

原创 2016年08月29日 07:24:57

题目描述

现在我们有一个int数组,请你找出数组中每个元素的下一个比它大的元素。
给定一个int数组A及数组的大小n,请返回一个int数组,代表每个元素比他大的下一个元素,若不存在则为-1。保证数组中元素均为正整数。
测试样例:

[11,13,10,5,12,21,3],7

返回:[13,21,12,12,21,-1,-1]

思路:从后往前维护一个单调栈,注意细节(代码把主函数加上了)

#include<cstdio>
#include<algorithm>
#include<vector>
#include<stack>
using namespace std;
vector<int> findNext(vector<int> A, int n) {
    // write code here
    vector<int>v(n);
    if(n==0) return v;
    stack<int>s;
    for(int i=n-1;i>=0;i--){
        if(s.empty()){
            s.push(A[i]);
            v[i]=-1;
        }
        else if(s.top()>A[i]){
            v[i]=s.top();
            s.push(A[i]);
        }
        else{
            bool ok=true;
            while(!s.empty()){
                if(s.top()>A[i]){
                    v[i]=s.top();
                    s.push(A[i]);
                    ok=false;
                    break;
                }
                s.pop();
            }
            if(ok){
                s.push(A[i]);//记得要放进去
                v[i]=-1;
            }
        }
    }
    return v;
}
int main(){
    int n,x;
    vector<int>v,ans;
    freopen("in.txt","r",stdin);
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d",&x);
        v.push_back(x);
    }
    ans=findNext(v,n);
    printf("n==%d\n",n);
    for(int i=0;i<n;i++){
        if(i&&i%10==0){
            printf("\n");
        }
        printf(" %d",ans[i]);
    }
    printf("\n");
    return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

[程序员面试金典]1002.下一个较大元素

题目描述现在我们有一个int数组,请你找出数组中每个元素的下一个比它大的元素。 给定一个int数组A及数组的大小n,请返回一个int数组,代表每个元素比他大的下一个元素,若不存在则为-1。保证数组中...

496. Next Greater Element I(寻找下一个较大的数)

You are given two arrays (without duplicates) nums1 and nums2 where nums1’s elements are subset of...

15、一个整数数列,元素取值可能是1~N(N是一个较大的正整数)中的任意一个数,相同数值不会重复出现。

/************************************************************************/ /* 15、一个整数数列,元素取值可能是1~N(N...

下一个较大值

数据结构实验之栈六:下一较大值(二) Time Limit: 150MS Memory limit: 8000K 题目描述 对于包含n(1 输入  输入有多组,第一行输入t(1...

程序员面试金典:下一个最大元素

1.下一个较大元素 2.下一个较大元素(2) 题目描述     现在我们有一个int数组,请你找出数组中每个元素的下一个比它大的元素。给定一个int数组A及数组的大小n,请返回一个int数组,代...

数据结构Stack实例(三):求元素的下一个更大的值

利用栈求数组元素下一个更大的值

某公司的一道面试题:最新版本号;输入两个表示版本号的字符串,返回版本较大的一个

请实现一个函数:最新版本号;输入两个表示版本号的字符串,返回版本较大的一个;版本字符串分为3断:主版本号,次版本号和编译版本号,之间用”.”分隔,前面的号更重要;比如 ”13.21” > ”5.6...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:下一个较大元素
举报原因:
原因补充:

(最多只允许输入30个字)