实现一个MyQueue类,该类用两个栈来实现一个队列。
示例:
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 操作)。 很简单的题目,进队列时将元素压入 inStack 中,出队列时将 inStack 元素顺序压入 outStack 后弹出顶端元素即可。
public class MyQueue {
private Stack<int> inStack, outStack;
/** Initialize your data structure here. */
public MyQueue() {
inStack = new Stack<int>();
outStack = new Stack<int>();
}
/** Push element x to the back of queue. */
public void Push(int x) {
Reverse(outStack, inStack);
inStack.Push(x);
}
/** Removes the element from in front of queue and returns that element. */
public int Pop() {
Reverse(inStack, outStack);
return outStack.Pop();
}
/** Get the front element. */
public int Peek() {
Reverse(inStack, outStack);
return outStack.Peek();
}
/** Returns whether the queue is empty. */
public bool Empty() {
return (inStack.Count | outStack.Count) == 0;
}
// 将 st1 中的元素压入 st2 中
private void Reverse(Stack<int> st1, Stack<int> st2) {
while (st1.Count != 0) st2.Push(st1.Pop());
}
}
/**
* Your MyQueue object will be instantiated and called as such:
* MyQueue obj = new MyQueue();
* obj.Push(x);
* int param_2 = obj.Pop();
* int param_3 = obj.Peek();
* bool param_4 = obj.Empty();
*/