1. 程式人生 > >劍指offer-Java實現:題目5、利用兩個棧實現佇列

劍指offer-Java實現:題目5、利用兩個棧實現佇列

題目描述

用兩個棧來實現一個佇列,完成佇列的Push和Pop操作。 佇列中的元素為int型別。

 

思路:棧的特點:先進後出 。 佇列特點:先進先出。自己畫了畫圖,看的比較秦楚,大概是每次進佇列操作都進stack1棧,出佇列時必須先把stack1彈棧到stack2中,這樣剛好就是倒過來的順序了。

stack1 :    1      >      2  1 >     3  2  1 > .

stack2 :    3      >      2   3  >     1  2   3 

所以出隊時就彈出最後面進入的元素。

最重要的是,每次彈完 為了保持佇列順序必須要 再 把stack2完全彈棧到stack1中

stack 1  :    2   >   3   2 .

stack 2  :    null;

 

程式碼:

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    int k=0;
    
    public void push(int node) {
        stack1.push(node);
        k++;
    }
    
    public int pop() {
        if(stack1.isEmpty()){
            return 0;
        }
        while(!stack1.isEmpty()){
            stack2.push(stack1.pop());
        }
        int x = stack2.pop();
        while(!stack2.isEmpty()){
            stack1.push(stack2.pop());
        }
        return x;
    }
}