正如标题所述,你需要使用两个栈来实现队列的一些操作。
队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。
pop和top方法都应该返回第一个元素的值。
样例
比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2
分析:就是用两个栈实现先进先出
class Queue {
public:
stack<int> stack1;
stack<int> stack2;
Queue() {
// do intialization if necessary
}
void push(int element) {
// write your code here
stack1.push(element);
}
int pop() {
// write your code here
if(stack2.empty())
{
while(!stack1.empty())
{
int data=stack1.top();
stack2.push(data);
stack1.pop();
}
}
int res=stack2.top();
stack2.pop();
return res;
}
int top() {
// write your code here
if(stack2.empty())
{
while(!stack1.empty())
{
int data=stack1.top();
stack2.push(data);
stack1.pop();
}
}
return stack2.top();
}
};