Skip to content

Latest commit

 

History

History
55 lines (40 loc) · 1.31 KB

File metadata and controls

55 lines (40 loc) · 1.31 KB

用两个队列实现栈

题目描述

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

解法

Push 操作,每次都存入 queue1Pop 操作,每次从 queue1 取:

  • queue1 中的元素依次倒入 queue2,直到 queue1 剩下一个元素,这个元素就是要 pop 出去的;
  • queue1queue2 进行交换,这样保证每次都从 queue1 中存取元素,queue2 只起到辅助暂存的作用。
import java.util.LinkedList;
import java.util.Queue;

/**
 * @author Anonymous
 * @since 2019/10/29
 */

public class Solution {

    private Queue<Integer> queue1 = new LinkedList<>();
    private Queue<Integer> queue2 = new LinkedList<>();

    public void push(int node) {
        queue1.offer(node);
    }

    public int pop() {
        if (queue1.isEmpty()) {
            throw new RuntimeException("Empty stack!");
        }

        while (queue1.size() > 1) {
            queue2.offer(queue1.poll());
        }

        int val = queue1.poll();

        Queue<Integer> t = queue1;
        queue1 = queue2;
        queue2 = t;
        return val;

    }
}

测试用例

  1. 往空的栈里添加、删除元素;
  2. 往非空的栈添加、删除元素;
  3. 连续删除元素直至栈为空。