牛客练习赛10 B栈和排序【思维】

题目描述

给你一个1->n的排列和一个栈，入栈顺序给定
你要在不打乱入栈顺序的情况下，对数组进行从大到小排序
当无法完全排序时，请输出字典序最大的出栈序列

输入描述:

第一行一个数n
第二行n个数，表示入栈的顺序，用空格隔开，结尾无空格

输出描述:

输出一行n个数表示答案，用空格隔开，结尾无空格

示例1

输入

5
2 1 5 3 4

输出

5 4 3 1 2

说明

2入栈；1入栈；5入栈；5出栈；3入栈；4入栈；4出栈；3出栈；1出栈；2出栈

备注:

对于100%的数据，有1<=n<=1000000，保证给的是一个排列

题意： 略

分析： 我们可以模拟下找到规律，因为输出字典序最大的，第一个输出的肯定是最大的元素，当考虑第i个元素时，如果当前元素是$[a[i],n]$,中最大的元素，那么就输出，否则就入栈，如果输出后，考虑栈顶元素，同上，扫一遍之后如果栈中还有元素，那么就直接输出即可

参考代码

#include<bits/stdc++.h>

using namespace std;

stack<int> s;
vector<int> res;
int a[N],pre[N];

int main(){
    ios_base::sync_with_stdio(0);
    int t = -1;
    int n;cin>>n;
    for(int i = 0;i < n;i++) cin>>a[i];
    for(int i = n-1;i >= 0;i--) {
        t = max(t,a[i]);
        pre[i] = t;
    }
    for(int i = 0;i < n;i++) {
        if(a[i] == pre[i]) {
            res.push_back(a[i]);
            while (!s.empty() && s.top() > pre[i+1]) {
                res.push_back(s.top());
                s.pop();
            }
        } else {
            s.push(a[i]);
        }
    }
    while (!s.empty()) {
        res.push_back(s.top());
        s.pop();
    }
    for(int i = 0;i < res.size();i++) {
        printf("%d%c",res[i],i == res.size() - 1 ? '\n' : ' ');
    }
    return 0;
}

• 如有错误或遗漏，请私聊下UP，thx