一:题目
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
示例 1:
输入:
["CQueue","appendTail","deleteHead","deleteHead","deleteHead"]
[[],[3],[],[],[]]
输出:[null,null,3,-1,-1]
示例 2:
输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]
二:解题思路
第一步:
用两个栈实现尾部插入和头部删除,首先我们就需要在全局定义两个栈,分别用作插入栈(inStack)和删除站(outStack)
stack<int> inStack, outStack;
第二步:实现插入
实现队尾的插入,直接运用push函数即可
void appendTail(int value)
{
inStack.push(value);
}
第三步:实现删除
第一种情况:栈为空
这种情况只需要判断两个栈为空,返回-1即可
if(outStack.empty())
{
if(inStack.empty())
{
return -1;
}
}
第二种情况:栈中有数字需要删除
假如,我们现在先后插入了5和2两个数字在我们的inStack里面
因为是从队头开始删除,所以我们是先删除5然后删除2
那么只需要将inStack里面的数据一个一个压入outStack的栈即可
总代码:
class CQueue {
private:
stack<int> inStack, outStack;//设置的两个栈,进栈和出栈
public:
CQueue() {
}
void appendTail(int value) {//直接push压入
inStack.push(value);
}
int deleteHead() {
if(outStack.empty())
{
if(inStack.empty())//都为空返回-1
{
return -1;
}
else
{
while(!inStack.empty())//将inStack的数据全部转入outStack中
{
outStack.push(inStack.top());
inStack.pop();
}
}
}
int value=outStack.top();//将outStack中的数据一个一个出去
outStack.pop();
return value;
}
};