牛客—栈和排序

传送门
 

题目描述

给你一个1->n的排列和一个栈，入栈顺序给定

你要在不打乱入栈顺序的情况下，对数组进行从大到小排序

当无法完全排序时，请输出字典序最大的出栈序列
 

输入描述:

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

输出描述:

输出一行n个数表示答案，用空格隔开，结尾无空格（其实结尾有空格也可以过，不过正式比赛还是以题意为准，不然罚时就难受了）

题意分析：根据栈的顺序（先入后出）把题目给出的顺序按照最大的排序顺序输出

但是怎么判断出栈的先后顺序是个问题，这个时候我们思考一下，因为是先入后出的顺序，那只要我们输入的这个数以后，后面的的数没有比这个数更大了，那是不是就是最大的结果了。

带着这个思路，我们开始写

这是第一遍的代码图，因为题目没有给n的范围，所以用双重for循环试了一下，只过了35%的案例，应该是超时了

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int a[1100000],b[110];
map<int, int>mp;
stack<int>sk;
int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        cin >> a[i];
    }
    for (int i = 0; i < n; i++)
    {
        sk.push(a[i]);
        int flag = 1;
        for (int j = i + 1; j < n; j++)
        {
            if (a[i] < a[j])
            {
                flag = 0;
                break;
            }
        }
        if (flag)
        {
            cout << sk.top() << " ";
            sk.pop();
        }
    }
    while (!sk.empty())
    {
        cout << sk.top() << " ";
        sk.pop();
    }
}

既然不能用双重for循环，那就换一种思路，只用一层for循环也可以搞定。因为是要找到后续数里面的最大值，我们可以从最后面开始往前比较，多开一个b[n]=0,然后用b[i]=max(a[i],b[i+1])去进行比较，则b[i]所储存的就是其后面的数的最大值。最后只需要比较这个最大值来判断是入栈还是出栈。

完善代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int a[1100000],b[1100000];
map<int, int>mp;
stack<int>sk;
int main()
{
	int n;
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		cin >> a[i];
	}
	b[n] = 0;
	for (int i = n - 1; i >= 0; i--)b[i] = max(b[i + 1], a[i]);
	for (int i = 0; i < n; i++)
	{
		sk.push(a[i]);
		while (!sk.empty() && sk.top() > b[i + 1])
		{
			cout << sk.top() << " ";
			sk.pop();
		}
	}
}