题目描述:
用两个栈来实现一个队列,完成push()和top()功能;
注:本系列程序都是在牛客网上进行编写测试通过,并且在VS2015环境下进行测试通过;
思路解读:
主要是将stack1作为主栈,将stack2作为辅栈,stack1作为push()压入队列的主要栈操作,如果是pop()弹出队列,则用stack2不断的接收stack1.top()栈顶的元素,压入栈中,这样直到把stack1中所有的元素都压入stack1中,此时,stack2则是一个反转的stack1,这时只需要记录stack2.top()元素就可以,并且将stack2栈顶元素弹出,然后此时再以相同的方法将stack2中的元素全部压入stack1中,这样stack1就获得了一个删除了第一个元素的新的栈,pop()功能就类似于queue中的元素弹出了;
代码如下:
//用两个栈来实现一个队列
//能实现队列的插入与弹出功能,并返回弹出的数
#include<stack>
#include<iostream>
using namespace std;
class Solution
{
public:
void push(int node) {
stack1.push(node);
}
int pop() {
while (!stack1.empty()) {
stack2.push(stack1.top());
stack1.pop();
}
int pop_value = stack2.top();
stack2.pop();
while (!stack2.empty()) {
stack1.push(stack2.top());
stack2.pop();
}
return pop_value;
}
private:
stack<int> stack1;
stack<int> stack2;
};
int main(int argc,char** argv) {
Solution test;
test.push(3);
test.push(13);
test.push(38);
cout << "Result:" << test.pop() << endl;
system("pause");
return 0;
}
结果如下: