路曼曼其修远兮,吾将上下而求索
问题描述: 用两个栈实现一个队列。实现它的两个函数 appendTail (完成在队列尾部插入整数功能)和 deleteHead(在队列头部删除整数的功能)
方法:一个栈存储元素,一个栈辅助
思路: 根据栈 “先进后出” 的特性,第一个栈的底部元素是最后插入的元素,第一个栈的顶部元素是下一个被删除的元素。为了维护队列的特性,每次插入的元素应该在第一个栈的底部。因此每次插入元素时,若第一个栈内已经有元素,应将已有的全部元素依次弹出并压入第二个栈,然后将新元素压入第一个栈,最后将第二个栈内的全部元素依次弹出并压入第一个栈。经过上述操作,新插入的元素在第一个栈的底部,第一个栈内的其余元素的顺序和插入元素之前保持一致。
删除元素时,若第一个栈非空,则直接从第一个栈内弹出一个元素并返回,若第一个栈为空,则返回 -1。
另外维护队列的元素个数,用于判断队列是否为空。初始元素个数为 0。每次插入元素,元素个数加 1。每次删除元素,元素个数减 1。
代码如下:
template <typename T> class CQueue{
public:
CQueue(void);
~CQueue(void);
void appendTail(const T & node);
T deleteHead();
private:
stack<T> stack1;
stack<T> stack2;
};
插入元素: appendTail
如果 stack1 非空,则将 stack1 内的元素依次弹出并依次压入 stack2,直至 stack1 内的全部元素都被弹出
将新元素 value 压入 stack1 内
如果 stack2 非空,则将 stack2 内的元素依次弹出并依次压入 stack1,直至 stack2 内的全部元素都被弹出
将 size 的值加 1
删除元素:deleteHead
如果 size 为 0,则队列为空,返回 -1
如果 size 大于 0,则队列非空,将 size 的值减 1,从 stack1 弹出一个元素并返回
class CQueue {
public:
stack<int> A;
stack<int> B;
int size = 0;
CQueue() {
}
void appendTail(int value) {
A.push(value);
size++;
}
int deleteHead() {
if(size == 0) return -1;
if (B.empty() && !A.empty()){
while(!A.empty()){
B.push(A.top());
A.pop();
}
}
int res = B.top();
B.pop();
size--;
return res;
}
};