232. 用栈实现队列

232. 用栈实现队列

有个基本一样的题目 剑指 Offer 09. 用两个栈实现队列

题目描述

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列的支持的所有操作（push、pop、peek、empty）：

实现 MyQueue 类：

• void push(int x) 将元素 x 推到队列的末尾
• int pop() 从队列的开头移除并返回元素
• int peek() 返回队列开头的元素
• boolean empty() 如果队列为空，返回 true ；否则，返回 false

说明：

• 你只能使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
• 你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。
  进阶：

进阶：

你能否实现每个操作均摊时间复杂度为 O(1) 的队列？换句话说，执行 n 个操作的总时间复杂度为 O(n) ，即使其中一个操作可能花费较长时间。

示例：

输入：
[“MyQueue”, “push”, “push”, “peek”, “pop”,“empty”]
[[], [1], [2],[], [], []]
输出：
[null, null, null, 1, 1, false]

解释：

MyQueue myQueue = new MyQueue();
myQueue.push(1); // queue is: [1]
myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue)
myQueue.peek(); // return 1
myQueue.pop(); // return 1, queue is [2]
myQueue.empty(); // return false

提示：

1 <= x <= 9
最多调用 100 次 push、pop、peek 和 empty
假设所有操作都是有效的 (例如，一个空的队列不会调用 pop 或者 peek 操作）

方法一：双栈

思路

将一个栈当作输入栈，用于压入 push 传入的数据；另一个栈当作输出栈，用于 pop 和 peek 操作。

每次 pop 或 peek 时，若输出栈为空则将输入栈的全部数据依次弹出并压入输出栈，这样输出栈从栈顶往栈底的顺序就是队列从队首往队尾的顺序。

C++:

class MyQueue
{
private:
	stack<int> inStack, outStack;
	void int2Out()
	{
		while (!inStack.empty())
		{
			outStack.push(inStack.top());
			inStack.pop();
		}
	}
public:
	/** Initialize your data structure here. */
	MyQueue()
	{
		while (!inStack.empty())//初始化堆栈，清空处理
		{
			inStack.pop();
		}
		while (!outStack.empty())
		{
			outStack.pop();
		}
	}

	/** Push element x to the back of queue. */
	void push(int x)
	{
		inStack.push(x);
	}

	/** Removes the element from in front of queue and returns that element. */
	int pop()
	{
		if (outStack.empty())
		{
			int2Out();
		}
		if (outStack.empty())
		{
			return -1;
		}
		int t = outStack.top();
		outStack.pop();
		return t;
	}

	/** Get the front element. */
	int peek()
	{
		if (outStack.empty())
		{
			int2Out();
		}
		return outStack.top();
	}

	/** Returns whether the queue is empty. */
	bool empty()
	{
		return inStack.empty() && outStack.empty();
	}
};

C语言：

typedef struct  
{
	int* stk;
	int stkSize;
	int stkCapacity;
}Stack;

Stack* stackCreate(int capacity)
{
	Stack* ret = malloc(sizeof(Stack));
	ret->stk = malloc(sizeof(int)*capacity);
	ret->stkSize = 0;
	ret->stkCapacity = capacity;
	return ret;
}

void stackPush(Stack* obj, int x)
{
	obj->stk[obj->stkSize++] = x;
}
void stackPop(Stack* obj)
{
	obj->stkSize--;//这里的写法有点出乎意料,直接把stkSize--就行了，stkSize里的数据在调用free时再被清除  
}
int stackTop(Stack* obj)
{
	return obj->stk[obj->stkSize - 1];
}
bool stackEmpty(Stack* obj)
{
	return obj->stkSize == 0;
}
void stackFree(Stack* obj)
{
	free(obj->stk);
	free(obj);
}
typedef struct 
{
	Stack* inStack;
	Stack* outStack;
}MyQueue;
MyQueue* myQueueCreate()
{
	MyQueue* ret = malloc(sizeof(MyQueue));
	ret->inStack = stackCreate(100);
	ret->outStack = stackCreate(100);
	return ret;
}
void in2Out(MyQueue* obj)
{
	while (!stackEmpty(obj->inStack))
	{
		stackPush(obj->outStack, stackTop(obj->inStack));
		stackPop(obj->inStack);
	}
}
void myQueuePush(MyQueue* obj,int x)
{
	stackPush(obj->inStack, x);
}
int myQueuePop(MyQueue* obj)
{
	if (stackEmpty(obj->outStack))
	{
		in2Out(obj);
	}
	int temp = stackTop(obj->outStack);
	stackPop(obj->outStack);
	return temp;
}
int myQueuePeek(MyQueue* obj)
{
    if(stackEmpty(obj->outStack))
    {
        in2Out(obj);
    }
	return stackTop(obj->outStack);
}
bool myQueueEmpty(MyQueue* obj)
{
	return stackEmpty(obj->inStack) && stackEmpty(obj->outStack);
}
void myQueueFree(MyQueue* obj)
{
	stackFree(obj->inStack);
	stackFree(obj->outStack);
    free(obj);
}

复杂度分析

时间复杂度：push 和 empty 为 O(1)，pop 和 peek 为均摊 O(1)。对于每个元素，至多入栈和出栈各两次，故均摊复杂度为 O(1)。

空间复杂度：O(n)。其中 n 是操作总数。对于有 n 次 push 操作的情况，队列中会有 n 个元素，故空间复杂度为 O(n)。