题目:火车进站
题目描述:
给定一个正整数N代表火车数量,0<N<10,接下来输入火车入站的序列,一共N辆火车,每辆火车以数字1-9编号。要求以字典序排序输出火车出站的序列号。
输入描述:
有多组测试用例,每一组第一行输入一个正整数N(0<N<10),第二行包括N个正整数,范围为1到9。
输出描述:
输出以字典序从小到大排序的火车出站序列号,每个编号以空格隔开,每个输出序列换行,具体见sample。
示例
输入:
3
1 2 3
输出:
1 2 3
1 3 2
2 1 3
2 3 1
3 2 1
题解:
看到这道题,我首先想到的就是一个数列的模拟进栈出栈问题。
首先说几句题外话:
n个数的进栈出栈问题、n个节点的二叉树的形状个数、n对括号的正确匹配个数、n个节点的二叉搜索树个数等等,这些都与组合数学中的卡特兰数有关。
卡特兰数的一般通项公式为:
言归正传,这道题不是让写火车进站出站顺序情况有几种,而是让写所有可能的出站顺序。所有我们就要把n个数的所有输出情况依次试一试。恰好题目让以字典序从小到大排序的火车出站序列号输出,所有我们为了方便,就使用STL库中的sort函数和next_permutation函数。next_permutation函数就是要以字典序从小到大输出全排列,而我们要对这个排列就行判断,判断是否满足出栈顺序,所以我们需要自己定义一个函数用来此判断,具体的详见代码。
代码:
#include<iostream>
#include<algorithm>
#include<stack>
using namespace std;
bool judge(int pop[],int n,int push[])//判断是否为正确的出栈顺序
{
if(push==NULL || pop==NULL ||n<=0)
{
return false;
}
stack<int> Stack;
int i=0,j=0;
for(i=0;i<n;i++)//依次把push中的数入栈
{
Stack.push(push[i]);
while(j<n && Stack.size()!=0 && pop[j]==Stack.top())//依次判断pop序列每个值是否与栈顶相等
{
Stack.pop();
j++;
}
}
return Stack.empty();
}
int main()
{
int i,n;
while(cin>>n)
{
int push[n];
int pop[n];
for(i=0;i<n;i++)
{
cin>>pop[i];
push[i]=pop[i];
}
sort(pop,pop+n);
do
{
if(judge(pop,n,push))
{
for(i = 0; i < n; i++)
{
cout<<pop[i]<<" ";
}
cout<<"\n";
}
} while(next_permutation(pop, pop+n));
}
}