解题方法:

  • 如下图所示,我们使用将push操作放到stack1中,将pop操作放到stack2中,当pop操作且stack2为空时,将stack1中所有元素转入stack2中即可。图解如下:

图片说明

代码如下:

class Solution
{
public:
    void push(int node) {
        stack1.push(node);
    }

    int pop() {        
        if(stack2.empty()){
            //pop时 stack2为空,将stack1中所有元素转入到stack2中
            while(!stack1.empty()){
                stack2.push(stack1.top());
                stack1.pop();
            }
        }        
        int ans=stack2.top();
        stack2.pop();
        return ans;
    }

private:
    stack<int> stack1;
    stack<int> stack2;
};

复杂度分析:

时间复杂度:对于每个元素的push和pop操作时间复杂度都为O(1),push显然,pop时由于每个元素都只进入stack2一次,并且只删除一次,所以均摊时间复杂度为O(1)。
空间复杂度:O(n),两个栈,每个栈的空间至多为n。