描述:
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
(题目来自牛客网)
用C++实现如下
class Solution
{
public:
//思路,使用两个stack来实现队列queue,完成push和pop功能。定义两个成员变量stack1和stack2供函数用
//push时,直接向stack2里面push;
//pop时,看stack1的情况,取top有结果直接给res,弹;没有则先把stack2里面的值全部转到stack1,再弹
void push(int node) { //push函数定义为void类型,直接往stack里面push值进去
stack2.push(node);
}
int pop() { //pop定义为int类型,弹出之后得到一个int型的值
int res;
if(!stack1.empty()) //如果stack1不为空,直接弹出来
{
res = stack1.top();
stack1.pop();
return res;
}
//如果stack1为空,进入下面流程
while(!stack2.empty()) //把stack2中全部元素全部转到stack1中
{
stack1.push(stack2.top());
stack2.pop();
}
res = stack1.top(); //最终还是从stack1中弹值出来
stack1.pop();
return res;
}
private:
stack<int> stack1;
stack<int> stack2;
};
纯手撕代码,如果觉得内容不错麻烦点个赞,后面陆续配上Top100算法题通俗易懂的讲解视频,可以花两个月时间完全掌握,进大厂不是梦,转行狗亲测!