题目描述

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否可能为该栈的弹出顺序。

假设压入栈的所有数字均不相等。

例如序列1,2,3,4,5是某栈的压入顺序，序列4,5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。

注意：若两个序列长度不等则视为并不是一个栈的压入、弹出序列。若两个序列都为空，则视为是一个栈的压入、弹出序列。

样例

输入：[1,2,3,4,5]
      [4,5,3,2,1]

输出：true

算法

(模拟)

1. 给定了2个数组，我们新建一个栈stk，来模拟二者的弹出过程。如果满足条件，则最后stk会为空。
2. 我一开始写的for循环，其实有点麻烦了。while就可以，for(int i; i )这个也可以用while(){i }代替，在while外面给定一个i=0的初值就可以。
3. 要开始习惯用auto x

时间复杂度

$O(n)$

C++ 代码

class Solution {
public:
    bool isPopOrder(vector<int> A, vector<int> B) {
        // if(A.empty() || B.empty()) return true;  // **1
        if(A.size() != B.size()) return false;

        stack<int> stk;

        int j = 0;
        for(int i = 0; i < A.size(); i ++)
        {
            stk.push(A[i]);

            /*
            if(A[i] == B[i])
                stk.pop();
            else continue;
            */
            while(stk.size() && stk.top() == B[j])
            {
                stk.pop();
                j ++;
            }
        }

        return stk.empty();
    }
};

C++ 代码(简化版)

class Solution {
public:
    bool isPopOrder(vector<int> A, vector<int> B) {
        // if(A.empty() || B.empty()) return true;  // **1
        if(A.size() != B.size()) return false;

        stack<int> stk;

        /*
        int j = 0;
        for(int i = 0; i < A.size(); i ++)
        {
            stk.push(A[i]);

            /*
            if(A[i] == B[i])
                stk.pop();
            else continue;
            */
            /*
            while(stk.size() && stk.top() == B[j])
            {
                stk.pop();
                j ++;
            }
        }
        */

        int i = 0;
        for(auto x : A)
        {
            stk.push(x);
            while(stk.size() && stk.top() == B[i])
            {
                i ++;
                stk.pop();
            }
        }

        return stk.empty();
    }
};