栈的压入、弹出序列-剑指Offer

栈的压入、弹出序列

题目描述

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

思路

1. 需要一个辅助栈
2. 如果下一个弹出栈的数字刚好是栈顶数字，那么直接弹出。若不是，把压栈序列中没有入栈的的数字压入辅助栈，直到把下一个需要弹出的数字压入栈顶为止。若所有数字都压入栈仍然没有找到下一个弹出的数字，那这个序列就不可能是弹出栈的序列

代码

import java.util.ArrayList;
import java.util.Stack;

public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
      boolean result = false;

      if (pushA != null && popA != null && pushA.length != 0 && popA.length != 0) {
    	  Stack<Integer> stack = new Stack<Integer>();
    	  int pushNext = 0; // 压入序列的指针
    	  int popNext = 0; // 弹出序列的指针
    	  while (popNext < popA.length) {
    		  while (stack.empty() || (stack.peek() != popA[popNext])) {
    // 若栈顶不是要弹出的元素，就继续压栈
    			  if (pushNext == pushA.length) {
    				  break;
    			  }
    			  stack.push(pushA[pushNext]);
    			  pushNext++;
    		  }
    		  if (stack.peek() != popA[popNext]) {
    // 遍历完了都没找到弹栈元素，那就是没有
    			  break;
    		  }
    // 找到之后栈顶弹出，继续找下一个
    		  stack.pop();
    		  popNext++;
    	  }
    	  if (stack.empty() && popNext == popA.length) {
    // 该弹出的元素都弹出了，弹栈序列也遍历完了，那就是了
    		  result = true;
    	  }
      }

      return result;
    }
}