版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/weixin_41923658/article/details/90081893
题目:用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
完整代码:
class Solution
{
public:
void push(int node) {
//将元素添加到stack1里
stack1.push(node);
}
int pop() {
//定义临时变量
int data;
//如果stack2里没有元素
if(stack2.size()<=0)
{
//对于stack1中的全部元素
while(stack1.size()>0)
{
//取栈顶元素
data=stack1.top();
//将栈顶元素取出
stack1.pop();
//将取出的元素压入到stack2中
stack2.push(data);
}
}
//至此,stack1中的元素全部被弹出
if(stack2.size()==0)
printf("stack1 is empty!");
//否则,从stack2弹出的元素即是队列元素
//取stack2 得栈顶元素
data=stack2.top();
// 将元素弹出
stack2.pop();
//返回删除元素
return data;
}
private:
stack<int> stack1;
stack<int> stack2;
};