面试题31:栈的压入、弹出序列
题目:输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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. 先假设有一个辅助栈stack;
2. 此时栈为空,分析弹出序列,第一个为4,那么应该将依次压入1,2,3,4,此时栈顶为4,从栈顶弹出4;
3. 第二个为5,此时栈顶为3,所以将压入序列中剩余数字依次压入栈,直到栈顶为5为止,此时栈中为1,2,3,5,弹出5;
4. 第三个为3,此时栈顶刚好为3,弹出3;
5. 第四个为2,此时栈顶刚好为2,弹出2;
6. 第五个为1,此时栈顶刚好为1,弹出1;
而对于弹出序列为4,3,5,1,2而言
1. 分析弹出序列,第一个为4,那么应该将1,2,3,4依次压入栈,栈顶为4,弹出4;
2. 第二个为3,此时栈顶刚好为3,弹出3;
3. 第三个为5,此时栈顶为2,将剩余数字入栈,直至5,此时栈中为1,2,5,弹出栈顶5;
4. 第四个为1,此时栈顶为2,且所有的数字都已经入栈,所以该序列不是一个有效的出栈序列;
总结: 如果下一个弹出的数字不是栈顶,则需要把压栈序列中还没有入栈的数字压入辅助栈,直到把下一个需要弹出的数字压入栈顶为止;如果所有的数字都压入栈顶仍然没有找到下一个弹出的数字,那么该序列不可能是一个有效的弹出序列。
自己写的low代码
import java.util.ArrayList;
import java.util.Stack;
public class Solution {
public boolean IsPopOrder(int [] pushA,int [] popA) {
if (pushA == null || popA == null || pushA.length == 0 || popA.length == 0) {
return false;
}
// 辅助栈;
Stack<Integer> stack = new Stack();
int i = 0;
int j = 0;
while (i < popA.length) {
if (!stack.isEmpty() && stack.peek() == popA[i]) {
stack.pop();
i++;
} else {
//提前终止循环;
if(j==pushA.length) {
break;
}
while (j < pushA.length) {
if(pushA[j] != popA[i]) {
stack.push(pushA[j]);
j++;
continue;
}
if (pushA[j] == popA[i]) {
stack.push(pushA[j]);
j++;
break;
}
}
}
}
if (i == popA.length) {
return true;
} else {
return false;
}
}
}
优雅的代码
链接:https://www.nowcoder.com/questionTerminal/d77d11405cc7470d82554cb392585106
来源:牛客网
import java.util.ArrayList;
import java.util.Stack;
public class Solution {
public boolean IsPopOrder(int [] pushA,int [] popA) {
if(pushA.length == 0 || popA.length == 0)
return false;
Stack<Integer> s = new Stack<Integer>();
//用于标识弹出序列的位置
int popIndex = 0;
for(int i = 0; i< pushA.length;i++){
s.push(pushA[i]);
//如果栈不为空,且栈顶元素等于弹出序列
while(!s.empty() &&s.peek() == popA[popIndex]){
//出栈
s.pop();
//弹出序列向后一位
popIndex++;
}
}
return s.empty();
}
}