PAT-ADVANCED1051/Data Structures and Algorithms7-3——Pop Sequence

我的PAT-ADVANCED代码仓：https://github.com/617076674/PAT-ADVANCED

我的Data Structures and Algorithms代码仓：https://github.com/617076674/Data-Structures-and-Algorithms

原题链接：

PAT-ADVANCED1051：https://pintia.cn/problem-sets/994805342720868352/problems/994805427332562944

Data Structures and Algorithms7-3：https://pintia.cn/problem-sets/16/problems/665

题目描述：

PAT-ADVANCED1051、Data Structures and Algorithms7-3：

题目翻译：

1051 弹出顺序

给定一个最多可以保留M个数的堆栈。按1，2，3，...，N的顺序将N个数字压入栈并随机弹出。 你需要判断给定的数字序列是否是堆栈的可能弹出序列。 例如，如果M为5、N为7时，我们可以从堆栈中获得1，2，3，4，5，6，7，但不能获得3，2，1，7，5，6，4。

输入格式：

每个输入文件包含一个测试用例。对于每个测试用例，第一行包含3个数字（全部不超过1000）：M（堆栈的最大容量），N（入栈序列的长度）和K（要检查的弹出序列的数量）。然后是K行，每行包含一个N个数字的弹出序列。一行中的所有数字都用空格分隔。

输出格式：

对于每个弹出序列，如果确实是堆栈的可能弹出序列，则在一行中打印“是”，否则打印为“否”。

输入样例：

5 7 5
1 2 3 4 5 6 7
3 2 1 7 5 6 4
7 6 5 4 3 2 1
5 6 4 3 7 2 1
1 7 6 5 4 3 2

输出样例：

YES
NO
NO
YES
NO

知识点：栈

思路：按顺序入栈，一旦栈顶元素与序列中当前元素相同即出栈

我们首先关注序列中的第一个元素，即令index = 0。

我们按1，2，3，4……的顺序依次入栈，一旦栈的大小超过了M，判断其为“NO”。

如果栈不为空，一旦栈顶元素的值和序列中索引为index的值相等，令栈顶元素出栈，且index++。

最后如果栈为空且入栈出栈过程中栈中的元素个数始终没有超过M，则输出“YES”，否则输出“NO”。

时间复杂度是O(K * N)。空间复杂度是O(N)。

C++代码：

#include<iostream>
#include<stack>

using namespace std;

int main() {
	int M, N, K;
	scanf("%d %d %d", &M, &N, &K);
	int nums[N];
	stack<int> numStack;
	for(int i = 0; i < K; i++) {
		while(!numStack.empty()) {
			numStack.pop();
		}
		for(int j = 0; j < N; j++) {
			scanf("%d", &nums[j]);
		}
		int index = 0;
		bool flag = true;
		for(int j = 1; j <= N; j++) {
			numStack.push(j);
			if(numStack.size() > M) {
				flag = false;
				break;
			}
			while(!numStack.empty() && numStack.top() == nums[index]) {
				index++;
				numStack.pop();
			}
		}
		if(flag && numStack.empty()) {
			printf("YES\n");
		} else {
			printf("NO\n");
		}
	}
}

C++解题报告：