Leetcode 用队列实现栈

Leetcode 用队列实现栈

题目描述

请你仅使用两个队列实现一个后入先出（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

解题代码

解这道题主要需要考虑的是队列是下“先进先出”，而栈是“先进后出” ，因此我用的方法是将一个队列用作主队列，另一个用作辅助队列，每当需要“将栈顶元素出栈”或“返回栈顶元素”时，将主队列中的元素按顺序出栈并存到辅助队列中，最后保存下来最后出队列的元素，这个元素即“出栈元素”。最后在将辅助队列中元素出队列并按顺序存入主队列中即可。进队列和判断队列是否为空都只需在主队列上进行。

class MyStack {
	queue<int> q1;
	queue<int> q2;
public:
	/** Initialize your data structure here. */
	MyStack() {

	}

	/** Push element x onto stack. */
	void push(int x) {
		q1.push(x);
	}

	/** Removes the element on top of the stack and returns that element. */
	int pop() {
		int tmp, i,num;
		int num1 = q1.size();
		//queuesize=q1.size();
		for (i = 1; i < num1; i++)
		{
			tmp = q1.front();
			q2.push(tmp);
			q1.pop();
		}
		num = q1.front();
		q1.pop();
		for (i = 1; i < num1; i++)
		{
			tmp = q2.front();
			q1.push(tmp);
			q2.pop();
		}
		return num;
	}

	/** Get the top element. */
	int top() {
		int tmp1, tmp2, i;
		int num = q1.size();
		// size=q1.size();
		for (i = 1; i <= num; i++)
		{
			tmp1 = q1.front();
			q2.push(tmp1);
			q1.pop();
		}

		for (i = 1; i <= num; i++)
		{
			tmp2 = q2.front();
			q1.push(tmp2);
			q2.pop();
		}
		return tmp1;
	}

	/** Returns whether the stack is empty. */
	bool empty() {
		return q1.empty();
	}
};

结果