栈的压入弹出序列（剑指offer-21）

Posted 2020-12-14 yzhengy

题目描述

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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
遍历压栈顺序，先将第一个放入栈中，这里是1，然后判断栈顶元素是不是出栈顺序的第一个元素，这里是4，很显然1≠4，所以我们继续压栈，
直到相等以后开始出栈，出栈一个元素，则将出栈顺序向后移动一位，直到不相等，这样循环等压栈顺序遍历完成，如果辅助栈还不为空，
说明弹出序列不是该栈的弹出顺序。最后辅助栈为空。如果不为空说明弹出序列不是该栈的弹出顺序。

题目解答

import java.util.*;
public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
      Stack<Integer> stack = new Stack<>();
      int pushindex = 0;
      int popindex = 0;
      
      while(pushindex < pushA.length){
          if(!stack.isEmpty() && stack.peek()==popA[popindex]){//栈顶元素等于出栈元素时，直接出栈
              stack.pop();
              popindex++;}
          else{
              stack.push(pushA[pushindex]);//栈顶元素不等于出栈元素时，再往栈里面放数据
              pushindex++;
          }
      }
        //while循环是为了防止当所有入栈的元素都压入栈的时候，栈顶元素和出栈序列的下标所指的数字没有来得及比较。
        while (!stack.isEmpty()) {
            if (stack.peek() == popA[popindex]) {
                stack.pop();
                popindex++;
            } else {
                return false;
            }
        }
        return true;
    }
}

简约写法

public boolean IsPopOrder(int[] pushSequence, int[] popSequence) {
    int n = pushSequence.length;
    Stack<Integer> stack = new Stack<>();
    for (int pushIndex = 0, popIndex = 0; pushIndex < n; pushIndex++) {
        stack.push(pushSequence[pushIndex]);
        while (popIndex < n && !stack.isEmpty() 
                && stack.peek() == popSequence[popIndex]) {
            stack.pop();
            popIndex++;
        }
    }
    return stack.isEmpty();
}