剑指offer:用两个栈来实现一个队列

题目:

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

思路:

stack1用来入队列,stack2用来出队列。每当有元素入队列时,直接入stack1;如果要出队列,首先判断stack2是否为空,如果不是,直接弹出stack2栈顶元素,如果是空栈,先将stack1中元素全部弹出到stack2,然后弹出stack2栈顶元素。

补充背景知识:stack的常用操作,stack.push(x),stack.pop(),stack.empty(),stack.top()

queue的常用操作,queue.push(x),queue.pop(),queue.empty(),queue.front(),queue.back(),queue.size()

代码:

在线测试OJ:

https://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6?tpId=13&tqId=11158&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

AC代码:

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

    int pop() {
        if(stack2.empty())
	{
		if(stack1.empty())
			return -10000; //栈顶为空还要弹出的错误情况
		else
		{
			while(!stack1.empty())
			{
				stack2.push(stack1.top());
				stack1.pop();
			}
		}
	}
        int temp = stack2.top();
	stack2.pop();
	return temp;
    }

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

扩展:用两个队列实现一个栈

思路:

方法1:当有元素入栈时,先入队列queue1。当有元素出栈时,先将queue1中前n-1个元素出队列并入队列2,然后让最后剩下的一个元素出队列,完成出栈,之后还需要将queue2中元素全部出队列并入queue1,以便下一元素入栈的顺序是正确的。这样做的缺点是元素需要反复进行出入队列操作,效率不高。

方法2:两个队列交替使用,当有元素入栈时,把它压入非空队列(起始两个队列都为空,任选一个入队列即可)。出栈时,将非空队列的元素依次出队列并压入另一个空队列,然后将剩下的最后一个元素弹出,完成出栈操作。

class MyStack {
public:
    /** Initialize your data structure here. */
    MyStack() {
        
    }
    
    /** Push element x onto stack. */
    void push(int x) {
        if(q1.empty()&&q2.empty())
            q1.push(x);
        else if(q1.empty()&&!q2.empty())
                 q2.push(x);
        else if(!q1.empty()&&q2.empty())
                 q1.push(x);
    }
    
    /** Removes the element on top of the stack and returns that element. */
    int pop() {
        if(q2.empty())
        {
            while(q1.size()>1)
            {
                q2.push(q1.front());
                q1.pop();
            }
            int temp=q1.front();
            q1.pop();
            return temp;
        }
        else
        {
            while(q2.size()>1)
            {
                q1.push(q2.front());
                q2.pop();
            }
            int temp=q2.front();
            q2.pop();
            return temp;
        }
        
    }
    
    /** Get the top element. */
    int top() {
        if(!q1.empty())
            return q1.back();
        else
            return q2.back();
    }
    
    /** Returns whether the stack is empty. */
    bool empty() {
        return (q1.empty()&&q2.empty());
    }
private:
    queue<int> q1;
    queue<int> q2;

};

/**
 * Your MyStack object will be instantiated and called as such:
 * MyStack obj = new MyStack();
 * obj.push(x);
 * int param_2 = obj.pop();
 * int param_3 = obj.top();
 * bool param_4 = obj.empty();
 */

猜你喜欢

转载自blog.csdn.net/u012991043/article/details/81017185