題目：用兩個棧實現隊列

用兩個棧實現一個隊列。隊列的聲明如下，請實現它的兩個函數 appendTail 和 deleteHead ，分別完成在隊列尾部插入整數和在隊列頭部刪除整數的功能。(若隊列中沒有元素，deleteHead?操作返回 -1 )

示例 1：

輸入：
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
輸出：[null,null,3,-1]

示例 2：

輸入：
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
輸出：[null,-1,null,null,5,2]

提示：

1 <= values <= 10000
最多會對 appendTail、deleteHead 進行 10000 次調用

---

模板：

class CQueue {
public:CQueue() {}void appendTail(int value) {}int deleteHead() {}
};/*** Your CQueue object will be instantiated and called as such:* CQueue* obj = new CQueue();* obj->appendTail(value);* int param_2 = obj->deleteHead();*/

解題：

入棧操作O(1)，出棧操作最壞O(n)，最好O(1)。

class CQueue {
private:stack<int> inStack;stack<int> outStack;
public:CQueue() {}void appendTail(int value) {inStack.push(value);}int deleteHead() {int res=-1;if(!outStack.empty()) {res=outStack.top();outStack.pop();} else if(!inStack.empty()) {while(!inStack.empty()) {outStack.push(inStack.top());inStack.pop();}res = outStack.top();outStack.pop();}return res;}
};

拓展：如果不要求棧的個數，可以用一個vector解決：

class CQueue {
private:vector<int> inStack;
public:CQueue() {}void appendTail(int value) {inStack.push_back(value);}int deleteHead() {int res =-1;if(inStack.size()!=0){res = inStack[0];inStack.erase(inStack.begin());}return res;}};

總結

以上是生活随笔為你收集整理的剑指offer：面试题09. 用两个栈实现队列的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。