剑指 Offer 09. 用两个栈实现队列

题目

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

输入：
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出：[null,null,3,-1]

输入：
[“CQueue”,“deleteHead”,“appendTail”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[],[5],[2],[],[]]
输出：[null,-1,null,null,5,2]

思路：
两个栈实现队列尾部插入和队列头部删除整数，首先分解功能，栈的特点后进先出；然后队列是先进先出，从这里可以知道队列头部就是栈的底部，队列尾部插入就是栈插入。而单个栈是做不到栈底删除，需要借助多一个外部空间才行。所以队列尾部插入用一个栈，然后队列头部删除的时候就把A栈所有元素放到B栈，然后B栈执行出栈，就能把A栈的底部元素删除。

python 版本

class CQueue(object):
	def __init__(self):
		self.A,self.B=[],[]
	def appendTail(self, value):
		self.A.append(value)
	
	def deleteHead(self):
	#如果A栈空，返回-1； 栈B满，出栈B；如果A不满，A栈出栈，B栈入栈
		if self.B return self.B.pop()
		if not self.A:
			return -1
		while(self.A):
			self.B.append(self.A.pop())
		return self.B.pop()
# Your CQueue object will be instantiated and called as such:
# obj = CQueue()
# obj.appendTail(value)
# param_2 = obj.deleteHead()

C++ 版本

class CQueue{
public:
	stack<int> stack1;
	stack<int> stack2;
	CQueue(){}

	void appendTail(int value){
		stack1.push(value);
	}
	
	int deleteHead(){
		if(stack1.empty()) return -1;
		while(!stack.empty()){
			int tmp=stack1.top();
			stack1.pop();
			stack2.push(tmp);
		}
		//delete head
		int res=stack2.pop();
		stack2.pop();
		while(!stack2.empty()){
			int temp=stack2.top();
			stack2.pop();
			stack1.push(temp);
		}
		return res;
	}

};