LeetCode刷题---栈---946
验证栈序列
946. 验证栈序列 - 力扣(LeetCode)
题目
给定 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 之前弹出。
提示:
1 <= pushed.length <= 1000
0 <= pushed[i] <= 1000
pushed
的所有元素 互不相同popped.length == pushed.length
popped
是pushed
的一个排列
自己的思路和代码
思路
就是把pushed按照popped的方式进行入栈,如果popped的元素不在栈中,那么就将pushed所找元素之前的全部压入栈中,如果在栈中,那就观察栈顶元素是不是所找的元素,如果不是就返回false。
代码
class Solution {
public:
bool validateStackSequences(vector<int>& pushed, vector<int>& popped) {
stack<int> st;
multiset<int> table;
int i = 0;
for(auto it : popped) {
if(table.count(it)) {
if(st.top() != it) {
return false;
} else {
st.pop();
table.erase(it);
}
} else {
for(i; i<pushed.size(); i++) {
if(pushed[i] != it) {
st.push(pushed[i]);
table.insert(pushed[i]);
} else {
++i;
break;
}
}
}
}
return true;
}
};