大学的数据结构考试就有一道这样的题:
1,2,3,4,5是栈的压入顺序,判断4,5,3,2,1和4,3,5,1,2是不是对应的一个弹出序列。
自己脑子里模拟栈来顺一遍的话很简单,可是今天我遇到了用代码实现这个判断,我变得十分无知…
当然,依旧是看讨论区大佬们的代码偷学~
解决这个问题的思路就是:
压入顺序:pushA:1,2,3,4,5
弹出顺序: popA:4,5,3,2,1
1.引用一个真正的栈s来作为中介
2.把pushA的第一个元素放入栈,用这个元素比较pushA中的第一个元素,如果不相等,就把pushA中的元素继续压入栈;如果相等,就把该元素弹出。接着把popA的第二个元素进行比对,
pushA | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
popA | 4 | ||||
栈操作 | s.push(1) | s.push(2) | s.push(3) | s.push(4);s.pop() |
栈s中:1 2 3 4➡1 2 3
pushA | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
popA | 5 | ||||
栈操作 | 出栈 | s.push(5);s.pop() |
栈s中:1 2 3 5➡1 2 3
pushA | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
popA | 3 | ||||
栈操作 | s.pop() | 出栈 | 出栈 |
栈s中:1 2 3 ➡1 2
pushA | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
popA | 2 | ||||
栈操作 | s.pop() | 出栈 | 出栈 | 出栈 |
栈s中:1 2 ➡1
pushA | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
popA | 1 | ||||
栈操作 | s.pop() | 出栈 | 出栈 | 出栈 | 出栈 |
栈s中:1 ➡空
pushA | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
popA | 4 | 5 | 3 | 2 | 1 |
栈操作 | 出栈 | 出栈 | 出栈 | 出栈 | 出栈 |
3.直到popA中元素全部比对结束,如果栈s已经为空,说明这是正确的顺序,如果栈不为空,说明这是错误的顺序。
public boolean IsPopOrder(int [] pushA,int [] popA) {
if(pushA.length!=popA.length) return false;
Stack<Integer> s=new Stack<Integer>();
int j=0;
for(int i=0;i<popA.length;i++){
s.push(pushA[i]);
while(j<popA.length&&popA[j]==s.peek()){
s.pop();
j++;
}
}
return s.isEmpty()?true:false;
}