天空之城

Posted 2020-08-29

题目描述 Description

天使城有一个火车站，每辆火车都从A方向驶入车站，
再从B方向驶出车站。

 

为了调度火车，火车站设有停放轨道，可存放5辆火车。已知从A进入车站顺序为1、2、3……。现在给你一个调度方案，判断是否可行，如果可行，输出出站顺序。
有以下几种调度方法：
A. 将A上的头一辆车驶出B方向
B. 将A上的头一辆车停入暂停轨道
C. 将暂停轨道上最外面的车驶出B方向

输入描述 Input Description

输入第一行一个整数N(n<30)表示调度方案步骤数目。
下一行一个字符串，有N个大写字母，表示调度方法。

输出描述 Output Description

输出若不可行（暂停站满了还停车、暂停站空了还出车），则输出一行“No”。
若可行，输出一行“Yes”，再输出若干行，每行一个整数，表示车出站序列。

样例输入 Sample Input

[样例输入1]
6
ABBCCA
[样例输入2]
5
BACAC

 

样例输出 Sample Output

[样例输出1]
Yes
1
3
2
4
[样例输出2]
No

 

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

int N,full,train,num; 
int queue[31],stack[31];
char s;

int main()
{
	cin>>N;
	for(int i=1;i<=N;i++)
	{
		cin>>s;
		if(s==‘A‘)
		{
			num++;
			train++;
			queue[train]=num;
		}
		if(s==‘B‘)
		{
			if(full==5)
			{
				cout<<"No"<<endl;
				return 0;
			}
			num++;
			full++;
			stack[full]=num;
		}
		if(s==‘C‘)
			if(full!=0)
			{
				train++;
				queue[train]=stack[full];
				full--;
			}
			else
			{
				cout<<"No"<<endl;
				return 0;
			}
	}
	cout<<"Yes"<<endl;
	for(int i=1;i<=num;i++)
		cout<<queue[i]<<endl;
return 0;
}