您的位置:首页 > 职场人生

剑指offer——面试题7:用两个栈实现队列

2017-10-29 10:39 344 查看
题目(牛客网):用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型

代码:

class Solution
{
public:
void push(int node) {
stack1.push(node);

}

int pop() {
int temp = 0;
if(stack2.empty())
{
while(!stack1.empty())
{
temp = stack1.top();
stack2.push(temp);
stack1.pop();
}
}
temp = stack2.top();
stack2.pop();
return temp;
}

private:
stack<int> stack1;
stack<int> stack2;
};
分析:用两个栈实现一个队列,也就是编写两个函数,一个实现在队列的尾部插入结点,一个实现在队列的头部删除结点。此时,利用两个栈来实现,比如元素 a b c d,先压入到栈1,顺序为:a ,b,c ,d。此时如果要按照队列那样,先进先出,先弹出来的是 a。此时可以将栈1的元素弹出,借助另外一个栈,将元素压入到栈2中。这样,将栈2的栈顶元素弹出,就是队列的弹出顺序了。

《剑指offer》一书的 P59,对此题有详细的描述。但是书上采用的编程形式与牛客网有很大不同,程序如下:

template <typename T> class CQueue
{
public:
CQueue(void);
~CQueue(void);

// 在队列末尾添加一个结点
void appendTail(const T& node);

// 删除队列的头结点
T deleteHead();

private:
stack<T> stack1;
stack<T> stack2;
};

template <typename T> CQueue<T>::CQueue(void)
{
}

template <typename T> CQueue<T>::~CQueue(void)
{
}

template<typename T> void CQueue<T>::appendTail(const T& element)
{
stack1.push(element);
}

template<typename T> T CQueue<T>::deleteHead()
{
if(stack2.size()<= 0)
{
while(stack1.size()>0)
{
T& data = stack1.top();
stack1.pop();
stack2.push(data);
}
}

if(stack2.size() == 0)
throw new exception("queue is empty");

T head = stack2.top();
stack2.pop();

return head;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: