面试题 03.06. 动物收容所

题目

动物收容所。有家动物收容所只收容狗与猫，且严格遵守“先进先出”的原则。在收养该收容所的动物时，收养人只能收养所有动物中“最老”（由其进入收容所的时间长短而定）的动物，或者可以挑选猫或狗（同时必须收养此类动物中“最老”的）。换言之，收养人不能自由挑选想收养的对象。请创建适用于这个系统的数据结构，实现各种操作方法，比如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

---

解题思路

1.使用队列，符合先进先出

思路：

• 【注意】编号是自增的，所以编号越小年龄越大！而且编号不可能相等！
• 因此判断dequeueAny的时候直接对编号进行判断就可以了

class AnimalShelf {

    Queue<Integer> catQueue;
    Queue<Integer> dogQueue;

    public AnimalShelf() {
        catQueue = new LinkedList<>();
        dogQueue = new LinkedList<>();
    }

    public void enqueue(int[] animal) {
        if (animal.length == 0) {
            return;
        }
        if (animal[1] == 0) {
            catQueue.add(animal[0]);
        } else {
            dogQueue.add(animal[0]);
        }
    }

    public int[] dequeueAny() {
        if (dogQueue.isEmpty() && catQueue.isEmpty()) {
            return new int[]{-1, -1};
        } else if (dogQueue.isEmpty() && !catQueue.isEmpty()) {
            return new int[]{catQueue.poll(), 0};
        } else if (!dogQueue.isEmpty() && catQueue.isEmpty()) {
            return new int[]{dogQueue.poll(), 1};
        } else {
            int num = 0;
            int space = 0;
            if (catQueue.peek() > dogQueue.peek()) {
                num = dogQueue.poll();
                space = 1;
            } else {
                num = catQueue.poll();
            }
            return new int[]{num, space};
        }

    }

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

    public int[] dequeueCat() {
        if (catQueue.isEmpty()) {
            return new int[]{-1, -1};
        }
        return new int[]{catQueue.poll(), 0};
    }
}

public class Test {
    public static void main(String[] args) {
        /**["AnimalShelf", "enqueue", "enqueue", "enqueue", "dequeueAny", "dequeueAny"]
         [[], [[0, 0]], [[1, 1]], [[2, 0]], [], []]*/
        AnimalShelf animal = new AnimalShelf();
        animal.enqueue(new int[]{0,0});
        animal.enqueue(new int[]{1,1});
        animal.enqueue(new int[]{2,0});
        System.out.println(Arrays.toString(animal.dequeueAny()));
        animal.dequeueAny();
    }
}

性能：