【剑指offer-JZ9】用两个栈实现队列

80 阅读1分钟

用两个队列实现一个栈的功能?要求给出算法和思路:

  • 入栈:将元素进队列A
  • 出栈:判断队列A中元素的个数是否为1,如果等于1,则出队列(输出),如果不等于1,队列A中的元素依次出队列并放入队列B,直到队列A中的元素留下一个,然后出队列(输出),再把队列B中的元素依次出队列并放回队列A。
import java.util.Stack;


/*
用两个栈实现一个队列的功能?要求给出算法和思路

入队: 将元素进栈A
出队: 判断栈B是否为空, 如果为空,则将栈A中所有元素pop, 并push进栈B中, 栈B出栈一个;

                     如果不为空,栈B直接出栈一个。
 */
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.pop());
            }
            return stack2.pop();
        } else {
            return stack2.pop();
        }
    }
}


题源:www.nowcoder.com/practice/54…