给定一个入栈序列和一个出栈序列,判断该出栈序列是否是由给定入栈序列按照栈的规则得到的。
解决方案
遍历每个出栈序列,并模拟一次出栈的过程,然后检查是否和给定的入栈序列相匹配。
- 创建一个空栈,并准备一个变量
pushIndex
用来指示当前入栈元素的下标。 - 对于每个出栈序列中的元素:
- 如果当前栈为空或者栈顶元素不等于当前出栈元素,则需要从入栈序列中找到该元素并入栈,直到栈顶元素等于当前出栈元素或者入栈序列已经遍历完。
- 当前栈顶元素等于当前出栈元素,则出栈。
- 如果所有出栈元素都匹配了入栈序列中的某个排列,且出栈后栈为空,则当前排列是合法的出栈序列。
代码实现
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
// 检查给定的排列是否是合法的出栈序列
bool isValidPopSequence(const vector<int>& pushSequence, const vector<int>& popSequence) {
stack<int> st;
int pushIndex = 0;
// 遍历出栈序列
for (int i = 0; i < popSequence.size(); ++i) {
// 如果当前栈为空或者栈顶元素不等于当前出栈元素,则需要从入栈序列中找到该元素并入栈
while (st.empty() || st.top() != popSequence[i]) {
// 如果入栈序列已经遍历完了,但是栈顶元素还不等于当前出栈元素,则当前排列不是合法的出栈序列
if (pushIndex >= pushSequence.size()) {
return false;
}
st.push(pushSequence[pushIndex]);
pushIndex++;
}
// 当前栈顶元素等于当前出栈元素,出栈
st.pop();
}
// 如果所有出栈元素都匹配了入栈序列中的某个排列,且出栈后栈为空,则当前排列是合法的出栈序列
return st.empty();
}
int main() {
int n, m;
cin >> n >> m;
vector<int> pushSequence(n);
for (int i = 0; i < n; ++i) {
cin >> pushSequence[i];
}
for (int i = 0; i < m; ++i) {
vector<int> popSequence(n);
for (int j = 0; j < n; ++j) {
cin >> popSequence[j];
}
if (isValidPopSequence(pushSequence, popSequence)) {
cout << "yes" << endl;
} else {
cout << "no" << endl;
}
}
return 0;
}