class Solution { public: void push(int node) { while (!stack2.empty()){ stack1.push(stack2.top()); stack2.pop(); } stack1.push(node); } int pop() { while (!stack1.empty()){ stack2.push(stack1.top()); stack1.pop(); } int ret = stack2.top(); stack2.pop(); return ret; } private: stack<int> stack1; stack<int> stack2; };
C++ 的 stack.pop() 不返回值,需要用 stack.top() 然后再用 pop。
用两个栈模拟队列,
- stack1的顶部代表队列尾部的access,需要在队列尾部push时,把stack2所有元素倒序转移到stack1;
- stack2的顶部代表队列头部的access,需要在队列头部pop时,把stack1所有元素倒序转移到stack2;