PAT-ADVANCED1051——Pop Sequence

版权声明:我的GitHub:https://github.com/617076674。真诚求星! https://blog.csdn.net/qq_41231926/article/details/83930042

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

原题链接:https://pintia.cn/problem-sets/994805342720868352/problems/994805427332562944

题目描述:

题目翻译:

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个数字的弹出序列。一行中的所有数字都用空格分隔。

输出格式:

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

扫描二维码关注公众号,回复: 4021667 查看本文章

输入样例:

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++解题报告:

猜你喜欢

转载自blog.csdn.net/qq_41231926/article/details/83930042