LeetCode 946. 验证栈序列

题目描述

给定 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. 0 <= pushed.length == popped.length <= 1000
2. 0 <= pushed[i], popped[i] < 1000
3. 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();
    }
};