【LeetCode刷题记录】27.动物收容所

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

一、题目描述：

题目来源：LeetCode-动物收容所

动物收容所。有家动物收容所只收容狗与猫，且严格遵守“先进先出”的原则

。在收养该收容所的动物时，收养人只能收养所有动物中“最老”（由其进入收容所的时间长短而定）的动物，或者可以挑选猫或狗（同时必须收养此类动物中“最老”的）。

换言之，收养人不能自由挑选想收养的对象。

请创建适用于这个系统的数据结构，实现各种操作方法，比如enqueue、dequeueAny、dequeueDog和dequeueCat。允许使用Java内置的LinkedList数据结构。

enqueue方法有一个animal参数，animal[0]代表动物编号，animal[1]代表动物种类，其中 0 代表猫，1 代表狗。

dequeue*方法返回一个列表[动物编号, 动物种类]，若没有可以收养的动物，则返回[-1,-1]。

示例1:

输入：

["AnimalShelf", "enqueue", "enqueue", "dequeueCat", "dequeueDog", "dequeueAny"]

[[], [[0, 0]], [[1, 0]], [], [], []]

输出：

[null,null,null,[0,0],[-1,-1],[1,0]]

示例2:

输入：

["AnimalShelf", "enqueue", "enqueue", "enqueue", "dequeueDog", "dequeueCat", "dequeueAny"]

[[], [[0, 0]], [[1, 0]], [[2, 1]], [], [], []]

输出：

[null,null,null,null,[2,1],[0,0],[1,0]]

说明:

收纳所的最大容量为20000

二、思路分析：

思路一：

建立两个队列，分别存储猫和狗。

dequeCat和dequeDog分别取对应的队首。

dequeAny则比较两个队首的编号，animal[0]待办动物编号。

animal[1]代表动物种类，0代表猫，1代表狗。

思路二：

用1个队列存放猫和狗，收养任意动物时先判空再返回

收养指定动物则遍历找到第一个出现的指定动物，从队列中删除后返回该动物

三、AC 代码：

思路一：

    class AnimalShelf {
        LinkedList<int[]> carQueen;
        LinkedList<int[]> dogQueue;

        public AnimalShelf() {
            carQueen = new LinkedList<>();
            dogQueue = new LinkedList<>();
        }
        
        public void enqueue(int[] animal) {
            if (animal[1] == 0) carQueen.addLast(animal);
            else if (animal[1] == 1) dogQueue.addLast(animal);

        }

        public int[] dequeueAny() {
            int[] headCat;
            if (!carQueen.isEmpty()) headCat = carQueen.getFirst();
            else if (!dogQueue.isEmpty()) return dogQueue.removeFirst();
            else return new int[]{-1, -1};

            int[] headDog;
            if (!dogQueue.isEmpty()) headDog = dogQueue.getFirst();
            else return carQueen.removeFirst();

            if (headCat[0] <= headDog[0]) return carQueen.removeFirst();
            else return dogQueue.removeFirst();

        }

        public int[] dequeueDog() {
            if (!dogQueue.isEmpty()) return dogQueue.removeFirst();
            else return new int[]{-1, -1};

        }

        public int[] dequeueCat() {
            if (!carQueen.isEmpty()) return carQueen.removeFirst();
            else return new int[]{-1, -1};
        }
    }

思路二：

   class AnimalShelf {
        Queue<int[]> dogAndCatQueue;
        public AnimalShelf() {
            dogAndCatQueue = new LinkedList<>();
        }

        public void enqueue(int[] animal) {
            dogAndCatQueue.offer(animal);
        }

        public int[] dequeueAny() {
            if (dogAndCatQueue.isEmpty()) {
                return new int[]{-1, -1};
            }

            return dogAndCatQueue.poll();
        }

        public int[] dequeueDog() {
            for (int[] an : dogAndCatQueue)
                if (an[1] == 1) {
                    dogAndCatQueue.remove(an);
                    return an;
                }
            return new int[]{-1, -1};
        }

        public int[] dequeueCat() {
            for (int[] an : dogAndCatQueue)
                if (an[1] == 0) {
                    dogAndCatQueue.remove(an);
                    return an;
                }
            return new int[]{-1, -1};
        }
    }