剑指offer 21.栈的压入、弹出序列 & leetcode 剑指 Offer 31. 栈的压入、弹出序列

21. 栈的压入、弹出序列

题目描述

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4,5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。（注意：这两个序列的长度是相等的）

思路：

定义一个栈，每次push 一个元素，

比较栈顶元素和弹出序列的当前元素是否相同，如果相同，弹栈，持续比较出栈，直到不相等为止。 如果最后栈中还有元素或者 popA 数组还没到末尾，则返回 false， 否则返回 true

import java.util.ArrayList;
import java.util.Stack;
public class Solution {
    // 新建一个栈，将数组A压入栈中，当栈顶元素等于数组B时，就将其出栈，当循环结束时，判断栈是否为空，若为空则返回true.
    public boolean IsPopOrder(int [] pushA,int [] popA) {
      // 特殊情况
        if(pushA == null || popA == null || pushA.length == 0 || popA.length == 0){
            return false;
        }
        
        Stack<Integer> stack = new Stack<Integer>();
        int j = 0;    // 指向popA数组
        for(int i = 0; i < pushA.length; i++){
          
            stack.push(pushA[i]);
            while(!stack.isEmpty() && stack.peek() == popA[j]){
                stack.pop();
                j++;
            }
        }
　　　　 // 如果最后栈中海油元素或者 popA 数组还没到末尾，则返回 false
       if(!stack.isEmpty() || j < popA.length){
           return false;
       }
       return true;
    }
}

 leetcode 运行时间为：3ms, 空间为：38.5MB

复杂度分析：

时间复杂度：虽然看起来有两个for循环，但是其实程序只是把pushed和poped数组分别遍历了一遍，所以时间复杂度为O(2n)

空间复杂度：借助了一个栈，栈的大小即是空间复杂度，栈最大为O(n), 即把pushed序列全部压入栈时，栈的大小。所以空间复杂度为O(n)