有一个死胡同,宽度刚好只能让一辆汽车通过,偏偏老有汽车开到死胡同来,这下麻烦了,最先开来的汽车要最后才能倒退出去。给定一个汽车开来的序列和一个可能的倒车出去的序列,请判断汽车能否都倒退出去,若能则输出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. (In Press)
代码长度限制16 KB
时间限制400 ms
内存限制64 MB
代码:
#include<iostream>
#include <memory.h>
#include <stack>
using namespace std;
int main()
{
int T;
cin >> T;
while(T--)
{
int n;
cin >> n;
int a[n],b[n];
for(int i = 0; i < n; i++) //开来的顺序
{
cin >> a[i];
}
for(int i = 0; i < n; i++) //倒出的顺序
{
cin >> b[i];
}
stack<int> s; //定义栈
//先按顺依次入栈,判断每个进入数与倒出的数是否匹配
//如果匹配就出栈,不匹配入栈下一个数再进行匹配,
//最后所有数匹配完了,栈不为空就是不行,为空就是行
int j = 0;
for(int i = 0; i < n; i++)
{
//cout << "i:" << i << endl;
s.push(a[i]);
//如果栈顶元素和倒车顺序匹配就出栈,并且开始比对下一个数
while(!s.empty() && b[j] == s.top())
{
s.pop();
//cout << "j:" << j << endl;
j++;
}
}
if(s.empty())
{
cout << "Yes"<< endl ;
}
else
{
cout << "No" << endl;
}
}
return 0;
}