劍指offer【05】- 用兩個棧實現隊列(java)

題目：用兩個棧實現隊列

考點：棧和隊列

題目描述：用兩個棧來實現一個隊列，完成隊列的Push和Pop操作。 隊列中的元素為int類型。

 

解題思路：每次psuh是時先將stack2清空放入stck1(保證選入的一定在棧底)，stack2始終是用來刪除的。在pop前，先將stack1中中的數據清空放入stack2（保存后入的在棧底），stack1始終用於push。

 

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        //向stack2 push時，先判斷Stack2是否為空，
        //如果不為空則將stack2的元素出棧,放進stack1中
        while(!stack2.isEmpty()){
             stack1.push(stack2.pop());
        }
        //stack2為空，則直接放入元素
        stack2.push(node);
    }
    
    public int pop() {
        //棧2元素出棧時先判斷棧1是否為空
        //如果不為空則將stack1的元素出棧,放進stack2中
        while(!stack1.isEmpty()){
            stack2.push(stack1.pop());
        }
        //棧1為空，此時棧2直接出棧
        return stack2.pop();
    }
}