PAT甲级 1051 Pop Sequence (25分)

1051 Pop Sequence (25分)

题目链接：PAT A 1051

题目大意：输入第一行给出三个数m，n，k分别代表栈能存储的最大元素个数，从栈里取出元素的个数，需要判断的组数，然后k行，每行n个数。题目设定按照从1到n的顺序向栈里插入元素，k行中的数据代表从栈中取出元素的顺序，要求判定这些取出序列是否能成立。比如1 2 3 4 5 6 7就成立，因为这代表插入1，取出1，插入2，取出2，插入3，取出3，插入4，取出4，插入5，取出5，插入6，取出6，插入7，取出7，故成立。而3 2 1 7 5 6 4就不成立，插入1 2 3之后取出顺序为3 2 1没问题，随后插入4 5 6 7，取出顺序是7 6 5 4而无法做到7 5 6 4，故不成立。需要注意的就是栈这种结构是先入后出，取元素只能从栈顶取。题目要求符合要求的数据输出YES，不符合要求的数据输出NO。

思路分析：用到STL中的stack，定义一个初始变量now，赋值为0，bool变量flag，赋值为true。随后用一个for循环向栈中插入数据，每插入一个数，判断此时栈中元素个数是否大于给定的最大栈容量m，若超出就把flag设置为false，并退出循环。如果未超出就写一个while循环，判断此时的栈是否不为空且栈顶元素是否等于v[now]（数组v是存储输入序列的），如果相等就把栈顶元素抛出，同时now++，直到栈顶元素不等于v[now]或者是栈为空就退出循环，之后继续向栈里添加元素。最后通过判断flag是否为true以及栈是否为空来确定输出YES还是NO。

AC代码：

#include<iostream>
#include<stack>
#include<vector>
using namespace std;
int main() {
	int n, m, k;
	scanf("%d %d %d", &m, &n, &k);
	vector<int> v(n);
	while(k--) {
		stack<int> s;
		for(int i = 0; i < n; i++)
			scanf("%d", &v[i]);
		int now = 0;
		bool flag = true;
		for(int i = 1; i <= n; i++) {
			s.push(i);
			if(s.size() > m) { //栈中元素超过栈最大容量，肯定不符合要求，退出循环 
				flag = false;
				break;
			}
			while(s.empty() != true && s.top() == v[now]) { //一定要判断栈是否为空，否则s.top()会报错 
				s.pop();
				now++;
			}
		}
		if(flag == true && s.empty() == true)
			printf("YES\n");
		else
			printf("NO\n");
	}
	return 0;
}