题目:用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:队列:先进先出。栈:先进后出。push操作的时候直接放入一个栈,pop操作的时候则从另一个栈取,那么另一个栈怎么存数据呢?当这个栈没有数据的时候,则从push栈取出所有数据,此时最先存进栈的数据回到这个pop栈的顶。
class Solution { public: void push(int node) { stack1.push(node); } int pop() { int tmp; if (stack2.empty()) { while (!stack1.empty()) { tmp = stack1.top(); stack1.pop(); stack2.push(tmp); } } if (stack2.empty()) { return -1; } tmp = stack2.top(); stack2.pop(); return tmp; } private: stack<int> stack1; stack<int> stack2; };