剑指offer之用两个栈实现队列

题目描述

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

解题思路

1. PUSH的时候一直往Stack1里放，POP的时候将Stack1的所有元素POP正好PUSH进Stack2（两次后进先出正好反成先进先出），POP Stack2队首，然后再将Stack2的元素放回Stack1，操作完成。
2. 评论区大神思路是我的优化版，在POP Stack2队首元素后，不用将剩余元素放回Stack1，再次POP只要先判断Stack2是否为空，如空则将Stack1的所有元素放入Stack2，否则直接POP Stack2队首元素（因为Stack2 POP的顺序已经是先进先出的顺序了）

链接：https://www.nowcoder.com/questionTerminal/54275ddae22f475981afa2244dd448c6
来源：牛客网
用两个栈实现一个队列的功能?要求给出算法和思路!
<分析>：
入队：将元素进栈A
出队：判断栈B是否为空，如果为空，则将栈A中所有元素pop，并push进栈B，栈B出栈；
如果不为空，栈B直接出栈。

---

用两个队列实现一个栈的功能?要求给出算法和思路!
<分析>：
入栈：将元素进队列A
出栈：判断队列A中元素的个数是否为1，如果等于1，则出队列，否则将队列A中的元素 以此出队列并放入队列B，直到队列A中的元素留下一个，然后队列A出队列，再把 队列B中的元素出队列以此放入队列A中。

Code

• 思路1

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

    int pop() {
        int cur, res;
        while(!stack1.empty()) {
            cur = stack1.top();
            stack1.pop();
            stack2.push(cur);
        }
        res = stack2.top();
        stack2.pop();
        while(!stack2.empty()) {
            cur = stack2.top();
            stack2.pop();
            stack1.push(cur);
        }
        return res;
    }

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

• 思路2

//链接：https://www.nowcoder.com/questionTerminal/54275ddae22f475981afa2244dd448c6
//来源：牛客网

class Solution
{
public:
    void push(int node) {
       stack1.push(node);
    }
    
    int pop() {
        int a;
        if(stack2.empty()){
            while(!stack1.empty()){
                a=stack1.top();
                stack2.push(a);
                stack1.pop();
            }
        }
        a=stack2.top();
        stack2.pop();
        return a;    
    }
    
private:
    stack<int> stack1;
    stack<int> stack2;
};

• java

import java.util.Stack;

public class Solution {
	Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
    	stack1.push(node);
    }
    
    public int pop() {
        if(stack2.empty()) {
            while(!stack1.empty()) {
                stack2.push(stack1.peek());
                stack1.pop();
            }
        }
    	int res = stack2.peek();
    	stack2.pop();
    	return res;
    }
}

总结

• 两个栈可以模拟实现队列，思路非常巧妙，自己还是需要多想想
• 两个队列也可模拟栈操作，push的时候进入原本就有数据的那个队列，pop的时候将有数据的那个队列一直pop到只剩一个元素，其他的都进入另一个队列。