java數據結構和算法(05)兩個棧實現隊列

• 用兩個棧來實現一個隊列，完成隊列的Push和Pop操做。 隊列中的元素爲int類型。完成以下代碼：

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        
    }
    
    public int pop() {
    
    }
}
複製代碼

• 思路：push操做直接壓入棧stack1，pop操做藉助stack2，取出最早入棧的元素，再重構stack1
• 代碼：

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();

    public void push(int node) {
        stack1.push(node);
    }

    public int pop() {
       while (!stack1.isEmpty()){
           stack2.push(stack1.pop());
       }
       int first=stack2.pop();
       while (!stack2.isEmpty()){
           stack1.push(stack2.pop());
       }
       return first;
    }
複製代碼