数据结构实验之栈七:出栈序列判定
Time Limit: 30ms Memory limit: 1000K 有疑问?点这里^_^
题目描述
给一个初始的入栈序列,其次序即为元素的入栈次序,栈顶元素可以随时出栈,每个元素只能入栈依次。输入一个入栈序列,后面依次输入多个序列,请判断这些序列是否为所给入栈序列合法的出栈序列。
例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个出栈序列,但4,3,5,1,2就不可能是该序列的出栈序列。假设压入栈的所有数字均不相等。
输入
第一行输入整数
n(1<=n<=10000)
,表示序列的长度。
第二行输入n个整数,表示栈的压入顺序。
第三行输入整数t(1<=t<=10)。
后面依次输入t行,每行n个整数,表示要判断的每一个出栈序列。
输出
对应每个测试案例输出一行,如果由初始入栈序列可以得到该出栈序列,则输出
yes
,否则输出
no
。
示例输入
5 1 2 3 4 5 2 4 5 3 2 1 4 3 5 1 2
示例输出
yes no
#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(!S.empty() && S.top() == b[j]) //出站序列先和栈顶元素比较 { j++; S.pop(); } else if(a[i] == b[j]) //再和原序列进行比较 { j++; i++; } else if(i < n) { S.push(a[i]); i ++; } else break; } if(S.empty()) { cout << "yes" << endl; } else { cout << "no" << endl; } } return 0; }