用棧實現佇列_學習記錄
阿新 • • 發佈:2018-12-02
用棧實現佇列_學習記錄
【題目描述】
使用棧實現佇列的下列操作:
- push(x) -- 將一個元素放入佇列的尾部。
- pop() -- 從佇列首部移除元素。
- peek() -- 返回佇列首部的元素。
- empty() -- 返回佇列是否為空。
示例:
MyQueue queue = new MyQueue(); queue.push(1); queue.push(2); queue.peek(); // 返回 1 queue.pop(); // 返回 1 queue.empty(); // 返回 false
說明:
你只能使用標準的棧操作 -- 也就是隻有 push to top
peek/pop from top
, size
, 和 is empty
操作是合法的。
你所使用的語言也許不支援棧。你可以使用 list 或者 deque(雙端佇列)來模擬一個棧,只要是標準的棧操作即可。
假設所有操作都是有效的 (例如,一個空的佇列不會呼叫 pop 或者 peek 操作)。
【演算法實現】
/** * @author Guozhu Zhu * @date 2018/10/20 * @version 1.0 * */ class MyQueue { public Stack<Integer> stack1 = new Stack<Integer>(); public Stack<Integer> stack2 = new Stack<Integer>(); public MyQueue() { } public void push(int x) { stack1.push(x); } public int pop() { if (stack1.isEmpty() && stack2.isEmpty()) { return -1; } if (stack2.isEmpty()){ while (!stack1.isEmpty()) { stack2.push(stack1.pop()); } } return stack2.pop(); } public int peek() { if (stack1.isEmpty() && stack2.isEmpty()) { return -1; } if (stack2.isEmpty()){ while (!stack1.isEmpty()) { stack2.push(stack1.pop()); } } return stack2.peek(); } public boolean empty() { return (stack1.isEmpty()) && (stack2.isEmpty()); } }