XTUOJ-1035-列车长的烦恼

1. 题目

Description
John是个小列车站的站长，每次列车在这里重新编组时他就很烦恼。因为站上只有一个人字形的编组轨道(如图)，所有的列车车厢都是从人字轨的右边依次进去，从左边出来。但有一些编组顺序John总编不出来，John怀疑有些编组顺序是不可能完成的，可John又找不出那些是顺序是可以编组出，那些不可以。请你写一个程序帮助John辨别哪些编组可以完成，哪些不能完成。

输入：
第一行是一个整数K，表示有多少个测试用例，以后每行一个测试用例，每行为n+1个整数，第一个整数为n 表示有多少节车厢，后面n个整数表示需要编组成的顺序。比如说3节车厢，按照1，2，3依次入轨编组，可以在左边形成1 2 3，1 3 2，2 1 3，2 3 1，321。
输出：
每行输出一个测试用例的结果。如果可以编组输出Yes，否则输出No。
Sample Input
2
3 3 1 2
4 1 2 3 4
Sample Output
No
Yes

2. 解法
   可以发现这道题的本质就是给定入栈序列，求不可能的出栈序列。我们可以通过维护一个栈来解决这个问题。用一个数组存储给定的出栈序列，先按照给定的入栈序列压栈，直到入栈元素和数组中首个元素相等，然后依次出栈，直到和数组存储的元素不相等。循环这个过程…………最后如果栈空，那么出栈序列是可能的，否则可能。我的代码如下。

#include <stack>
#include <iostream>
using namespace std;
int a[10001];
int main(int argc, char const **argv) {
	int K;
	//freopen("D:\\myCCode\\XTUOJ\\test.txt", "r", stdin);
	cin >> K;
	while(K--) {
		stack<int> stack;
		int n;
		cin >> n;
		for(int i = 0; i < n; i++) {
			cin >> a[i];
		}
		for(int i = 0, j = 0; i < n; i++) {
			stack.push(i+1);
			while(!stack.empty() && stack.top() == a[j]) {
				stack.pop();
				j++;
			}
		}
		if(stack.empty()) {
			cout << "Yes" << endl;
		} else {
			cout << "No" << endl;
		}
	}
	return 0;
}