LeetCode Implement Queue using Stacks （数据结构）

 

 

题意：

　　用栈来实现队列。

 

思路：

　　一个栈是不够的，至少要两个。

　　（1）插入。永远只插入到stack1中（插到栈顶）。

　　（2）弹出。如果stack2不为空，直接弹出stack2的栈顶，否则，将stack1中的所有元素转移到stack2中，栈顶自然就是队头了，再弹出。

　　（3）返回队头。与（2）一样。

　　（4）是否为空。判断两个栈是否同时为空即可。

 

　　只要保证stack2为空时才可以将stack1转移到stack2中，就可以保证两边并不会产生混乱而出错。

 

class Queue {
/*
    // Push element x to the back of queue.
    void push(int x) {
        
    }

    // Removes the element from in front of queue.
    void pop(void) {
        
    }

    // Get the front element.
    int peek(void) {
        
    }

    // Return whether the queue is empty.
    bool empty(void) {
        
    }
*/  
    stack<int> stack1,stack2;
public:
    bool change()//当stack2为空时，将stack1转到stack2中
    {
        while(!stack1.empty())
        {
            stack2.push(stack1.top());
            stack1.pop();
        }
        if(stack2.empty())    return true;
        else    return false;
    }    
    void push(int x)
    {
        stack1.push(x);
    }
    void pop(void)
    {
        if(!stack2.empty())    stack2.pop();
        else if(!change())    stack2.pop();
    }
    int peek(void)
    {
        if(!stack2.empty())    return stack2.top();
        else
        {
            if(!change())    return stack2.top();
            return 0;
        }
    }
    bool empty(void)
    {
        if(stack1.empty()&&stack2.empty())    return true;
        else    return false;
    }
};