洛谷第一题

给出两个序列 pushed 和 poped 两个序列,其取值从 1 到 n(n\le100000)n(n100000)。已知入栈序列是 pushed,如果出栈序列有可能是 poped,则输出 Yes,否则输出 No。为了防止骗分,每个测试点有多组数据。

输入格式

第一行一个整数 qq,询问次数。

接下来 qq 个询问,对于每个询问:

第一行一个整数 nn 表示序列长度;

第二行 nn 个整数表示入栈序列;

第二行 nn 个整数表示出栈序列;

输出格式

对于每个询问输出答案。

输入输出样例

输入 #1
2
5
1 2 3 4 5
5 4 3 2 1
4
1 2 3 4
2 4 1 3
输出 #1
Yes
No


解题思路
我们输入两个数组,这两个数组元素相同,顺序相反。
将一个数组进栈,如果栈顶元素和b数组第一个元素相等则将该元素出栈。
最后判断栈是否为空,若为空则输出YES,若不为空则输出NO.

下面是代码
#include<iostream>
#include<stack>
using namespace std;
stack<int>q;//栈q 
int p,n;//p组数据,n为序列长度 
int main()
{
	cin>>p;
	while(p--)
	{
		cin>>n;
		int a[n+1],b[n+1],sum=1;//入栈队列a,待检验队列b,计数器sum 
		for(int i=1;i<=n;i++)
		cin>>a[i];
		for(int i=1;i<=n;i++)
		cin>>b[i];
		for(int i=1;i<=n;i++)
		{
			q.push(a[i]);//入栈 
			while((q.top())==b[sum])//当栈顶元素与b中当前元素相同时出栈 
			{
				q.pop(),sum++;//sum++到b下一个元素 
				if(q.empty())break;//注意这里,第一次RE就因为当栈空时还用了出栈操作,所以要手动结束循环 
			}
		}
		if(q.empty()) cout<<"Yes"<<endl;//如果栈为空说明出栈序列b正确 
		else cout<<"No"<<endl;
		while(!q.empty())q.pop();//清空栈 
	}
	return 0;//谢幕 
}
posted @ 2022-04-03 16:14  冉思银  阅读(83)  评论(0)    收藏  举报