火车进站

题目描述

给定一个正整数N代表火车数量,0<N<10,接下来输入火车入站的序列,一共N辆火车,每辆火车以数字1-9编号。要求以字典序排序输出火车出站的序列号。 

输入描述:

 

有多组测试用例,每一组第一行输入一个正整数N(0<N<10),第二行包括N个正整数,范围为1到9。

输出描述:

 

输出以字典序从小到大排序的火车出站序列号,每个编号以空格隔开,每个输出序列换行,具体见sample。

示例1

链接:https://www.nowcoder.com/questionTerminal/97ba57c35e9f4749826dc3befaeae109
来源:牛客网
 

思路:  用栈模拟火车进出站:对进站序列排序得到字典序的第一组,处理完第一组,用 next_permutation获取下一组排列。

对每组排列判断是否正确。在函数 isOutNum中实现:依次把进站序列压入栈,如果栈顶和出战序列相同就删除栈顶,不相同就继续压入进站序列。直到进站序列压入完毕。栈空表示所有元素可以顺利出栈,此组序列是正确的出栈序列。

#include<iostream>
#include<stack>
#include<algorithm>
using namespace std;
bool isOutNum(int *push,int *pop,int len)//判断pop是不是push的出栈序列
{
    if(push==NULL || pop==NULL ||len<=0)
        return false;
    stack<int> Stack;
    int i=0,j=0;
    for(i=0;i<len;i++)//依次把push中的数入栈
    {
        Stack.push(push[i]);
        while(j<len && Stack.size()!=0 && pop[j]==Stack.top())//依次判断pop序列每个值是否与栈顶相等
        {
            Stack.pop();
            j++;
        }
    }
    return Stack.empty();
}
int main()
{
    int N;
    while(cin>>N)
    {
        int *pushNum=new int [N];
        int *popNum=new int [N];
        for(int i=0;i<N;i++)
        {
            cin>>pushNum[i];
            popNum[i]=pushNum[i];
        }
        sort(popNum,popNum+N);
        do
        {
            if(isOutNum(pushNum,popNum,N))//如果该排列正确,则输出
            {
                for(int i=0;i<N-1;i++)
                    cout<<popNum[i]<<" ";
                cout<<popNum[N-1]<<endl;
            }
        }
        while(next_permutation(popNum,popNum+N));//获取下一个排列       
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/zwz2011303359/article/details/81085078