牛客 栈和排序 C++

题目描述

给定一个从 1 到 n 的排列 P，以及一个空栈。你按顺序将排列中的元素依次入栈，可以在任意时刻选择将栈顶元素出栈并将其加入输出序列。入栈顺序不可改变。

理想情况下，你想得到一个严格从大到小排序的输出序列 n,n−1,...,1，但受栈操作限制可能无法实现。当无法完全排序时，请输出**字典序**最大的合法出栈序列。

输入描述:

在一行中输入一个整数 n (1≦n≦10 6 )。 第二行输入 n 个整数，表示排列 P 中的元素，用空格分隔。保证给出的是一个从 1 到 n 的排列。

输出描述:

输出一行，包含若干整数，表示最终的出栈序列，用空格分隔，结尾不输出多余空格。

示例1

输入

5

2 1 5 3 4

输出

5 4 3 1 2

算法思路

就是在保证合法出栈的前提下，先输出较大的数。按顺序入栈，如果当前元素(栈顶元素)比后面的元素都大，就出栈。

代码

 #include<bits/stdc++.h>
using namespace std;
const int N=1000010;
int a[N],b[N];// a[]存储原始排列，b[]存储从i到n-1的最大值
int main()
{
	int n;
	cin>>n;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
	}
	b[n-1]=a[n-1];// b[i]表示从位置i到末尾的所有元素中的最大值
	for(int i=n-2;i>=0;i--)
	{
		b[i]=max(b[i+1],a[i]);
	}
	stack<int> s;
	for(int i=0;i<n;i++)
	{
		//当栈不为空且栈顶元素大于等于后面所有元素的最大值时
        //就可以得到字典序最大的输出
		if(!s.empty()&&s.top()>=b[i])
		{
			cout<<s.top()<<" ";
			s.pop();
		}
		s.push(a[i]);
	}
	// 处理栈中剩余元素(按出栈顺序输出)
	while(!s.empty())
	{
		cout<<s.top()<<" ";
		s.pop();
	}
	return 0;
}