题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路
建立一个辅助栈,把输入的第一个序列中的数字依次压入该辅助栈,并按照第二个序列的顺序依次从该栈中弹出数字。
- 如果下一个弹出的数字刚好是栈顶数字,那么直接弹出;
- 如果下一个弹出的数字不在栈顶,则把压栈序列中还没有入栈的数字压入辅助栈,直到把下一个需要弹出的数字压入栈顶为止;
- 如果所有数字都压入栈后仍然没有找到下一个弹出数字,那么该序列不可能是一个弹出序列。
解法1
import java.util.ArrayList; import java.util.Stack; public class Solution { public boolean IsPopOrder(int [] pushA,int [] popA) { if (pushA.length != popA.length || pushA.length == 0) return false; Stack<Integer> stack = new Stack<>(); int popIndex = 0; // 弹出序列的索引 for (int i = 0; i < pushA.length; i++){ stack.push(pushA[i]); // 判断栈是否为空的条件要放在前面,否则 最后为空时stack.peek()会报错 while (!stack.isEmpty() && stack.peek() == popA[popIndex]){ stack.pop(); popIndex++; } } return stack.isEmpty(); } }
解法2
import java.util.ArrayList; import java.util.Stack; public class Solution { public boolean IsPopOrder(int [] pushA,int [] popA) { if (pushA.length != popA.length || pushA.length == 0) return false; Stack<Integer> stack = new Stack<>(); stack.push(pushA[0]); //辅助栈初始化 int index = 1; for (int i = 0; i < popA.length; i++){ while (stack.peek() != popA[i] && index < pushA.length){ stack.push(pushA[index++]); } if (stack.peek() == popA[i]){ stack.pop(); continue; } return false; } return true; } }