给定 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 的排列。
一、原理
此题在字节跳动的笔试中出现过,不过换了一种说法,题目是说给定两个序列A和B,问是否可以经过push和pop操作将A变成B。字节的笔试对题意表达得更清楚一些。那么如何来做这道题呢?首先我们需要一个堆栈 s s s吧。那么我们下面需要考虑如何利用堆栈的push和pop操作得到另一个数组呢?对于堆栈而言首先要进行push不然何谈pop,所以应该按顺序将A中的元素push进去,当然不是一直push,需要找到停止push的条件。如果当前的栈顶元素恰好为B数组中未pop的首元素,应该立即将栈顶元素pop出来,因为如果不这样,再进行push操作,顺序一定是错的,这本质上是一种贪心。
算法
将 pushed
队列中的每个数都 push 到栈中,同时检查这个数是不是 popped
序列中下一个要 pop 的值,如果是就把它 pop 出来。
最后,检查不是所有的该 pop 出来的值都是 pop 出来了。
代码
class Solution {
public:
bool validateStackSequences(vector<int>& pushed, vector<int>& popped) {
stack<int> s;
int k=0;
for(auto num:pushed)
{
s.push(num);
while(!s.empty() && k<popped.size() && s.top()==popped[k])
{
k++;
s.pop();
}
}
return s.empty();
}
};