今天和大家聊的问题叫做 用队列实现栈,我们先来看题面:https://leetcode-cn.com/problems/implement-stack-using-queues/
Implement a last in first out (LIFO) stack using only two queues. The implemented stack should support all the functions of a normal queue (push, top, pop, and empty).
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通队列的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
- void push(int x) 将元素 x 压入栈顶。
- int pop() 移除并返回栈顶元素。
- int top() 返回栈顶元素。
- boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。
注意:
你只能使用队列的基本操作 —— 也就是 push to back、peek/pop from front、size 和 is empty 这些操作。你所使用的语言也许不支持队列。你可以使用 list (列表)或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
示例
输入: ["MyStack", "push", "push", "top", "pop", "empty"] [[], [1], [2], [], [], []] 输出: [null, null, null, 2, 2, false] 解释: MyStack myStack = new MyStack(); myStack.push(1); myStack.push(2); myStack.top(); // 返回 2 myStack.pop(); // 返回 2 myStack.empty(); // 返回 False
解题
因为栈和队列的存储方式是完全相反的,所以这里用到两个队列。
当有新元素要push时,用一个队列暂时存储之前的栈,另一个队列pop空,先压入新元素,再把暂时存储之前的栈的队列再一次push进队列中。
这样,队列中的元素顺序就是栈结构的顺序了。
class MyStack { public: queue<int> q1,q2; /** Initialize your data structure here. */ MyStack() { } /** Push element x onto stack. */ void push(int x) { while(!q2.empty()) { q1.push(q2.front()); q2.pop(); } q2.push(x); while(!q1.empty()) { q2.push(q1.front()); q1.pop(); } } /** Removes the element on top of the stack and returns that element. */ int pop() { int a = q2.front(); q2.pop(); return a; } /** Get the top element. */ int top() { return q2.front(); } /** Returns whether the stack is empty. */ bool empty() { return q2.empty(); } };
好了,今天的文章就到这里,如果觉得有所收获,请顺手点个在看或者转发吧,你们的支持是我最大的动力 。