解题思路
使用辅助栈 sta 来模拟 压入push/弹出pop操作的排列
关键逻辑
入栈操作: 按照压栈序列的顺序执行。
出栈操作: 每次入栈后,循环判断 “栈顶元素 === 弹出序列的当前元素” 是否成立,将符合弹出序列顺序的栈顶元素全部弹出。
算法流程:
1、使用辅助栈 sta, 并且把弹出序列中的第一个弹出的元素下标为 cur = 0
2、遍历操作:元素入栈,判断辅助栈的栈顶与弹出序列当前指针对应的元素是否相同
3、如果相同,则弹出该元素,并把当前元素的下一元素设为当前元素,即.++ cur,重复该过程直到栈为空或者栈顶元素不等于 popped 的当前元素;如果不相等,则回到第 2 步。
class Solution {
public:
bool validateStackSequences(vector<int>& pushed, vector<int>& popped)
{
stack<int> sta; // 辅助栈
int cur = 0; // 弹出序列中的当前的元素下标
for (int i = 0; i < pushed.size(); ++ i) {
sta.push(pushed[i]); // 入栈操作:将元素遍历插入辅助栈中
// 如果弹出序列中的当前元素等于辅助栈的栈顶元素,同时满足辅助栈不为空,则执行出栈操作,并且当前元素移到下一元素
while (!sta.empty() && sta.top() == popped[cur]) {
sta.pop();
++ cur;
}
}
return sta.empty(); // 如果辅助栈为空,返回true;否则,返回false
}
};