1. 程式人生 > >劍指offer面試題22 棧的壓入、彈出序列

劍指offer面試題22 棧的壓入、彈出序列

解題思路:

遍歷出棧序列,看棧中是否存在當前元素,若不存在,則在入棧序列中一直入棧,直到當前元素入棧為止。若存在,且是棧頂元素,則直接出棧;若存在,但不是棧頂元素,則證明此時的出棧序列是錯誤的。

import java.util.Stack;

public class Solution {
	
	public static boolean IsPopOrder(int [] pushA,int [] popA) {
		
		if(pushA == null || popA == null ||
			pushA.length == 0 || popA.length == 0) {
			return false;
		}
		
		boolean result = true;
		
		Stack<Integer> stack = new Stack<>();
		int nextStartIndex = 0;
		//遍歷出棧序列,看棧中是否存在當前元素,若不存在,則在入棧序列中一直入棧,直到當前元素入棧為止
		//若存在,且是棧頂元素,則直接出棧;若存在,但不是棧頂元素,則證明此時的出棧序列是錯誤的
		for (int i = 0; i < popA.length; i++) {
			int cur = popA[i];
			if (!stack.contains(cur)) {
				//如果棧中不存在當前元素,則從入棧序列的下一次起始位置開始入棧,直到cur入棧為止
				while (!stack.contains(cur)  && nextStartIndex < pushA.length) {
					stack.push(pushA[nextStartIndex]);
					nextStartIndex++;
				}
				
				if (stack.peek() == cur) {
					//入棧完成後,如果棧頂元素為當前元素,則出棧
					stack.pop();
				} else {
					//證明入棧序列中不存在當前元素
					return false;
				}
				
			} else  if (stack.peek() == cur){
				//如果棧中存在當前元素,並且當前元素是棧頂元素,則直接出棧
				stack.pop();
			} else {
				//如果棧中存在當前元素,但不是棧頂元素,則證明出棧序列不正確
				return false;
			}
			
			
		}
		return result;      
    }	
}