用兩個棧實現佇列 java
阿新 • • 發佈:2018-11-08
用兩個棧實現佇列 java
題目描述
用兩個棧來實現一個佇列,完成佇列的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) { stack1.push(node); } public int pop() { if(stack1.empty() && stack2.empty()){ throw new RuntimeException("Queue is empty"); } if(stack2.empty()){ while(!stack1.empty()){ stack2.push(stack1.pop()); } } int i = stack2.pop(); return i; } }