完整代码地址
题目
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。
例如序列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,1不等于4(弹出序列首个元素);
继续压入2,2不等于4;
继续压入3,3不等于4;
继续压入4;
此时4等于弹出序列的首个元素,将4弹出。
此时栈顶3不等于5(弹出序列的第二个元素),继续压入5;
此时5等于弹出序列的第二个元素,将5弹出。
此时3等于弹出序列的第三个元素,将3弹出。
此时2等于弹出序列的第四个元素,将2弹出。
此时1等于弹出序列的第五个元素,将1弹出。
此时压入序列遍历完毕,弹出序列也遍历完毕,栈为空,所以该弹出序列和该压入序列是匹配的。
代码
/**
* 输入两个整数序列,第一个序列表示栈的压入顺序,
* 请判断第二个序列是否为该栈的弹出顺序。
* 假设压入栈的所有数字均不相等。
*
* 例如序列1,2,3,4,5是某栈的压入顺序,
* 序列4,5,3,2,1是该压栈序列对应的一个弹出序列,
* 但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
*
* 测试:
* 1.两个数组都只含有一个数字(匹配或不匹配)
* 2.两个数组都含有多个数字(匹配或不匹配)
* 3.输入null
*
* @author peige
*/
public class _31_StackPushPopOrder {
public static boolean IsPopOrder(int [] pushA,int [] popA) {
if(pushA == null || popA == null || pushA.length != popA.length)
return false;
LinkedList<Integer> list = new LinkedList<>();
int popIndex = 0;
for(int i = 0; i < pushA.length; ++i) {
list.addFirst(pushA[i]);
if(pushA[i] == popA[popIndex]) {
list.removeFirst();
++popIndex;
while(!list.isEmpty()) {
if(list.peekFirst() == popA[popIndex]) {
list.removeFirst();
++popIndex;
continue;
}
break;
}
}
}
if(popIndex == popA.length && list.isEmpty())
return true;
return false;
}
}
测试
测试用例看上面的代码注释