下一个较大元素

题目描述

现在我们有一个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;}