题目描述
给定 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
的排列。
解题思路
顺序遍历pushed序列,遍历过程中与popped序列进行比较,如果相等,则都往后移动一位,如果不相等,先判断栈是否为空,如果不为空并且栈顶元素等于popped当前元素,则出栈同时移动到popped序列的下一位,否则将pushed当前元素进栈。
bool validateStackSequences(vector<int>& pushed, vector<int>& popped) {
int lpush = pushed.size(),lpop = popped.size();
if(lpush != lpop) return false;
stack<int> sk;
int i = 0,j = 0;
while(i < lpush){
if(pushed[i] == popped[j]){
i++;
j++;
}else{
if(!sk.empty() && sk.top() == popped[j]){
j++;
sk.pop();
}
else{
sk.push(pushed[i]);
i++;
}
}
}
while(!sk.empty() && j<lpop){
if(sk.top() != popped[j]) return false;
sk.pop();
j++;
}
return true;
}