题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:栈——先进后出;队列——先进先出
1、若s2不为空,则返回顶部元素;
2、若s2为空,s1不为空,则将s1元素出栈到s2中,返回s2顶部元素;
2、若s2为空,s1也为空,则返回-1.
class Solution
{
public:
void push(int node) {
stack1.push(node);
}
int pop() {
if(stack2.empty()){ //s2为空,s1不为空
while(!stack1.empty()){
stack2.push(stack1.top());
stack1.pop();
}
}
if(stack2.empty()) //s2为空,s1也为空
return -1;
int res = stack2.top(); //s1不为空时,出栈
stack2.pop();
return res;
}
private:
stack<int> stack1;
stack<int> stack2;
};
题目描述
用两个队列实现一个栈
#include<iostream>
#include<queue>
using namespace std;
class Solution
{
public:
void Push(int node) {
if (!q1.empty()) //保证其中一个队列为空,向不为空的队列中插入数据
q1.push(node);
else
q2.push(node);
}
int Pop() {
int ret = 0;
if (!q1.empty()) {
while (q1.size()>1)
{
q2.push(q1.front());
q1.pop();
}
ret = q1.front();
q1.pop();
}
else
{
while (q2.size()>1)
{
q1.push(q2.front());
q2.pop();
}
ret = q2.front();
q2.pop();
}
return ret;
}
private:
queue<int> q1;
queue<int> q2;
};
void Test() {
Solution s;
s.Push(1);
s.Push(2);
s.Push(3);
s.Push(4);
cout << s.Pop() << endl;
cout << s.Pop() << endl;
s.Push(5);
s.Push(6);
cout << s.Pop() << endl;
}
int main() {
Test();
return 0;
}