点击打开题目链接
#include <bits/stdc++.h>
using namespace std;
int a[10010], b[10010];//原序列,新序列
int main()
{
int n, k;
cin >> n;
for(int i = 0; i < n; i++)
{
scanf("%d", &a[i]);
}
cin >> k;
while(k --)
{
for(int i = 0; i < n; i++)
{
scanf("%d", &b[i]);
}
stack<int>S;//中间栈
int i = 0, j = 0;
while(j < n)
{
if(a[i] == b[j])
{
j++;
i++;
}
else if(!S.empty() && S.top() == b[j])//取栈顶元素和当前元素想比,若相等,将栈顶元素出栈
{
j++;//新序列元素更新
S.pop();
}
else if(i < n)
{
S.push(a[i]);//原序列入栈
i ++;
}
else break;
}
if(S.empty())//若中间栈为空,则此序列合法
{
cout << "yes" << endl;
}
else//否则不合法
{
cout << "no" << endl;
}
}
return 0;
}
假设原序列为12345,新序列为54321; 假设原序列为12345,新序列为43512;
原序列全部入栈; 前面几项均不匹配
栈中元素:12345,栈顶元素5,新序列元素5; 则1234入栈,栈中元素:1234,栈顶元素4,新序列元素4
栈中元素:1234,栈顶元素4,新序列元素4; 栈中元素:123,栈顶元素3,新序列元素3
栈中元素:123,栈顶元素3,新序列元素3; 栈中元素:125,栈顶元素5,新序列元素5
栈中元素:12,栈顶元素2,新序列元素2; 栈中元素:12,栈顶元素2,新序列元素1
栈中元素:1,栈顶元素1,新序列元素1; 此时,i不再满足条件,跳出循环
栈空 栈不空