剑指Offer09用两个栈实现队列

Posted 2022-12-13 hesorchen

题目

剑指 Offer 09. 用两个栈实现队列

用两个栈实现队列。

解题思路

感觉有$O(n)$的解法，但是想了好一会还是没想到，看了数据范围1e4，交了一发$O(n^2)$的过了。

原思路就是简单的来回倒。

题解思路

维护两个栈，一个维护插入，一个维护删除。需要删除的时候可以把插入栈里的元素全部倒进删除栈里，注意只有删除栈为空了，才会倒，否则会破坏元素的顺序。很巧妙。

代码

class CQueue

    stack<int> st1, st2;

public:
    CQueue()
    
        while (st1.size())
            st1.pop();
        while (st2.size())
            st2.pop();
    
    void appendTail(int value)
    
        st1.push(value);
    
    int deleteHead()
    
        if (!st2.size()) //为空才会倒，才能保证顺序正确
            while (st1.size())
            
                st2.push(st1.top());
                st1.pop();
            
        if (!st2.size())
            return -1;
        int res = st2.top();
        st2.pop();
        return res;
    
;