题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
示例 1:
输入:pushed = [1,2,3,4,5], popped = [4,5,3,2,1]
输出:true
解释:我们可以按以下顺序执行:
push(1), push(2), push(3), push(4), pop() -> 4,
push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1
示例 2:
输入:pushed = [1,2,3,4,5], popped = [4,3,5,1,2]
输出:false
解释:1 不能在 2 之前弹出。
提示:
0 <= pushed.length == popped.length <= 1000
0 <= pushed[i], popped[i] < 1000
pushed 是 popped 的排列。
LeetCode:栈的压入,弹出序列
解题思路
建立一个辅助栈,把压栈序列中的数字依次压入栈中(这里的压入并非一次性压入全部,而是以弹出序列中下一个要弹出的数字,详见代码注释),并按照弹出序列中的数字顺序依次从该栈中弹出数字
代码
public boolean validateStackSequences(int[] pushed, int[] popped) {
boolean flag = false;
// 建立一个辅助栈,把压栈序列中的数字依次压入栈中,并按照弹出序列中的数字顺序依次从该栈中弹出数字
if (pushed.length == 0 || pushed == null)
return true;
Stack<Integer> dataStack = new Stack<Integer>();
int length = pushed.length;
// 从 pushed 数组的第一个元素开始压入
int pPush = 0;
// 从 popped 数组的第一个元素开始判断
int pPop = 0;
for (int i = 0; i < length; i++) {
// 如果栈为空或者栈顶元素不是要弹出的元素,则把压栈序列中还没有入栈的数字压入 dataStack 中
while (dataStack.isEmpty() || dataStack.peek() != popped[pPop]) {
// 当压栈序列压入全部数字后,停止压入
if (pPush == length)
break;
// 压入 压栈序列中未压入 dataStack 的数字
dataStack.push(pushed[pPush]);
pPush++;
}
// 压栈序列中的数字都已经压入 dataStack 中,但栈顶元素与下一个要弹出的数字仍不等(while 循环 执行了 break 语句 或者 找不到符合弹出条件的栈顶元素),说明不是压栈序列的弹出序列,跳出 for 循环
if (dataStack.peek() != popped[pPop])
break;
// 如果上述 if 语句未执行,说明栈顶元素为下一个要弹出的数字,将栈顶元素从栈中弹出
dataStack.pop();
// 继续判断下一个要弹出的数字是否符合题意
pPop++;
}
// 如果栈为空且此时 pPop 指针到达了弹出序列的末尾,则表示匹配成功
if (dataStack.isEmpty() && pPop == length)
flag = true;
return flag;
}