【剑指Offer-31】栈的压入、弹出序列
问题
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。
示例
输入: pushed = [1,2,3,4,5], popped = [4,5,3,2,1]
输出: true
解释: 我们可以按以下顺序执行:
push(1), push(2), push(3), push(4), pop() -> 4,
push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1
解答
class Solution {
public:
bool validateStackSequences(vector<int>& pushed, vector<int>& popped) {
stack<int> s;
for (int i = 0, j = 0; i < pushed.size(); i++) {
s.push(pushed[i]);
while (!s.empty() && s.top() == popped[j]) {
s.pop();
j++;
}
}
return s.empty();
}
};
重点思路
使用一个栈来模拟压入和弹出的过程。当压入的值与j
指向的弹出的值相同时,对模拟栈执行弹出操作,直到栈空或者不满足上述条件为止。遍历整个压入序列,最后根据栈是否为空来判断第二个序列是否为对应的弹出顺序。