UVA514 Rails

铁轨 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<=1000$

输入输出格式

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

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

样例 #1

样例输入 #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

分析

这道题考察栈这种数据结构,总体算法不难,图中的station区域可以用一个栈结构来进行抽象.

#include <iostream>
#include <stack>
using namespace std;

int N = 0;

bool check(int* nums);
int main(int argc, char **argv){
	while(true){
		cin >> N;
		if(N == 0){
			break;
		}
		int nums[N];
		bool flag = true;
		while(flag){
			for(int i = 0; i < N; i++){
				cin >> nums[i];
				if(nums[0] == 0){
					flag = false;
					break;
				}
			}
			if(flag){
				cout << (check(nums) ? "Yes" : "No") << endl;	
			}
		}
		cout << endl;
	}
	return 0;
}
bool check(int* nums){
	stack<int> C;
	int p = 0;
	int i = 0;
	while(p < N){
		if(i == nums[p]){
			p++;
			i++;
		}else if(i < nums[p]){
			C.push(i);
			i++;
		}else{
			while(!C.empty() && C.top() == nums[p]){
				p++;
				C.pop();
			}
			if(!C.empty() && C.top() > nums[p]){
				return false;
			}
		}
	}
	return true;
}

做这种多组数据输入的时候,输出格式不容易看,所以可以使用条件编译,在调试的时候输出到文件中,这样更方便调试和查看