用兩個棧來實現(xiàn)一個隊列,完成隊列的Push和Pop操作。 隊列中的元素為int類型。
固定使用Stack1來壓棧
固定使用Stack2來彈棧
壓棧時,直接壓入Stack1中,
彈棧的時候,若Stack2中為空,則將Stack1全部彈棧壓入,即得到棧反轉(zhuǎn)后隊列
若Stack2不為空,則說明仍有部分前序隊列沒彈棧完畢,繼續(xù)彈棧
模擬場景(棧頂在右)
| 操作 | Stack1 | Stack2 | |
|---|---|---|---|
| A,B,C入棧 | A,B,C | ||
| 彈棧1次 | (彈棧前) | C,B,A | |
| 彈棧1次 | (開始彈棧) | C,B | |
| D入棧 | D | C,B | |
| 彈棧一次 | D | C | |
| E,F入棧 | D,E,F | C | |
| 彈棧一次 | D,E,F | ||
| 彈棧一次 | (彈棧前) | F,E,D | |
| 彈棧一次 | (開始彈棧) | F,E |
時間復雜度 壓棧O(1) 彈棧O(n)
具體實現(xiàn):
import java.util.Stack;
public class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
//無腦壓入
stack1.push(node);
return;
}
public int pop() {
//預存隊列已空,更新
if(stack2.empty()){
//stack1中內(nèi)容全部轉(zhuǎn)入stack2中作為隊列
while(!stack1.empty()){
stack2.push(stack1.pop());
}
}
//彈棧頂(隊列頭)出來
return stack2.pop().intValue();
}
}