洛谷-UVA514 铁轨 Rails

题目描述

输入格式

输出格式

题意翻译

某城市有一个火车站，铁轨铺设如图。有n节车厢从A方向驶入车站，按进站的顺序编号为1～n。你的任务是判断是否能让他们按照某种特定的顺序进入B方向的铁轨并驶出车站。例如，出栈顺序（5 4 1 2 3）是不可能的，但（5 4 3 2 1）是可能的。为了重组车厢，你可以借助中转站C。这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须按照相反的顺序驶出C。对于每节车厢，一旦从A移入C，就不能返回A了;一旦从C移入B，就不能返回C了。也就是说，在任意时刻，只有两种选择：A到C和C到B。

对于每一组数据，第一行是一个整数 N。接下来若干行数据，每行 N 个数，代表 1 ~ N 车厢的出栈顺序，最后一组数据只有一个整数 0 。对于每一组数据，在最后输出空行。

最后一组数据的 N=0 ，不输出。

n<=1000

输入输出样例

输入 #1复制

5
1 2 3 4 5
5 4 1 2 3
0
6
6 5 4 3 2 1
0
0

输出 #1复制

Yes
No
Yes

标准答案：

#include<bits/stdc++.h>
using namespace std;
const int maxsize=1000+5;
int n,B[maxsize];
int main()
{
    while(cin>>n&&n)
    {
        while(1)
		{
			int i=1,j=1;
			cin>>B[1];
			if(!B[1])
				break;
	        for(int i=2;i<=n;i++)
	        	cin>>B[i];
	        stack<int>s;
	        while(i<=n)
	        {
	            if(i==B[j])
	            {
	                i++;
	                j++;
	            }
	            else
	                s.push(i++);
	            while(!s.empty()&&s.top()==B[j])
				{
					j++;
					s.pop();
				}
			}
			if(j<=n)
				cout<<"No"<<endl;
	        else
				cout<<"Yes"<<endl;
		}
		cout<<endl;
    }
    return 0;
}

自己的答案：

#include <bits/stdc++.h>
using namespace std;

int main(){
	int n;
	int A[1000];
	int B[1005];
	while(cin>>n&&n!=0){
		while(true){
			cin>>B[0];
			if(B[0]==0){
				break;
			}
			//B出栈序号 
			for(int i=1;i<n;i++){
				cin>>B[i];
			}

			//A依次进站 
			for(int i=0;i<n;i++){
				A[i]=i+1;
			}
		
			//模拟A出栈、B进栈(也就是出栈顺序)
			int b=0; //B栈初始下标 
			stack<int> stack; //C中转栈	
				
			for(int i=0;i<n;i++){
				//A出栈(可能入C，可能入B)
				if(A[i]!=B[b]){
					stack.push(A[i]);
				}else{
					b++;				
				}
				//判断是否可从C出栈到B 
				while(!stack.empty()&&stack.top()==B[b]){
					stack.pop();
					b++;
				}					
			}
			
			//判断C栈是否为空
			if(stack.empty()){
				cout<<"Yes"<<endl;
			} else{
				cout<<"No"<<endl;
			}				
		}	
		cout<<endl;//不加这里测试用例不通过		
	}	
	return 0;
}