PTA 7-3 胡同(C++)

有一个死胡同，宽度刚好只能让一辆汽车通过，偏偏老有汽车开到死胡同来，这下麻烦了，最先开来的汽车要最后才能倒退出去。给定一个汽车开来的序列和一个可能的倒车出去的序列，请判断汽车能否都倒退出去，若能则输出Yes，否则输出No。

输入格式:

首先输入一个整数T，表示测试数据的组数，然后是T组测试数据。每组测试数据首先输入一个正整数n(n≤10)，代表开来的汽车数，然后输入2n个整数，其中，前n个整数表示汽车开来的序列，后n个整数表示汽车可能倒出的序列。

输出格式:

对于每组测试，判断能否倒车出该死胡同，若能则输出Yes，否则输出No。

输入样例:

2
4 1 2 3 4 2 1 4 3
4 1 2 3 4 4 2 1 3

输出样例:

Yes
No

提示:

样例1的可能方案为：1进，2进，2出，1出，3进，4进，4出，3出。

来源：

[1] 黄龙军,等. 数据结构与算法, 上海:上海交通大学出版社, 2022.7. ISBN: 9787313269881
[2] 黄龙军, 等. 数据结构与算法（Python版）,上海: 上海交通大学出版社, 2023. ISBN: 9787313280732

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

分析

这个题目中涉及到了栈中元素出栈顺序可行性判断的问题。可以采用将元素逐个入栈，每个元素入栈后与出栈顺序的首个元素比对，反复判断是否出栈的方法来进行解答。 最后结果的可行与否由栈是否为空决定。

示例代码

#include<iostream>
#include<stack>
using namespace std;
stack<int> cars;//用于存放汽车序号的栈

int main()
{
	int T;
	cin >> T;

	while (T--)//需要分析多组数据
	{
		int n = 0;
		int  order[22] = {0};

		cin >> n;
		for (int i = 0; i < 2 * n; i++)
			cin >> order[i];

		int  k = n;//order数组的0~n-1位置是入胡同顺序，n~2*n-1是出胡同顺序，k=n是出胡同数据的头号元素
		for (int j = 0; j < n; j++)
		{
			cars.push(order[j]); //将入胡同顺序元素逐个入栈
			while (!cars.empty() && cars.top() == order[k])
			{
				cars.pop();
				k++;   //对入栈的元素与出栈元素逐个对比，如相同，则出栈后再次检查是否可以出栈；如不同，进入下一次for循环
			}
		}

		if (cars.empty())//因为上面for循环中，所有车的号码都进栈了，所以cars如果为空则证明全部出栈，即后面给的n个数据的出栈顺序是可行的
			cout << "YES" << endl;
		else
			cout << "NO" << endl;
	}
	return 0;
}