题目描述
给定 pushed
和 popped
两个序列,只有当它们可能是在最初空栈上进行的推入 push 和弹出 pop 操作序列的结果时,返回 true
;否则,返回 false
。
示例 1:
输入: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
示例 2:
输入:pushed = [1,2,3,4,5], popped = [4,3,5,1,2] 输出:false 解释:1 不能在 2 之前弹出。
提示:
0 <= pushed.length == popped.length <= 1000
0 <= pushed[i], popped[i] < 1000
pushed
是popped
的排列。
问题分析
此题用一个空栈s模拟一下。遍历数组pushed,将每个遍历到的元素push进栈s,在push之后检查如果栈不为空并且栈顶元素等于popped_id指针指向的popped数组的元素,那么将s的栈顶弹出,然后将popped_id++。循环判断直到栈s为空或者栈顶元素不等于popped_id指针指向的元素时退出while循环。如此进行,当数组pushed遍历结束后,如果栈s为空,那么返回true,否则返回false。
代码实现
class Solution {
public:
bool validateStackSequences(vector<int>& pushed, vector<int>& popped) {
if(pushed.empty() && popped.empty())
return true;
if(pushed.empty() || popped.empty() || pushed.size() != popped.size())
return false;
stack<int> s;
int popped_id = 0;
for(int i = 0; i < pushed.size(); i++){
s.push(pushed[i]);
while(!s.empty() && s.top() == popped[popped_id]){
s.pop();
popped_id++;
}
}
return s.empty();
}
};