【程序员面试金典】面试题 03.06. 动物收容所

题目描述

描述：动物收容所。有家动物收容所只收容狗与猫，且严格遵守“先进先出”的原则。在收养该收容所的动物时，收养人只能收养所有动物中“最老”（由其进入收容所的时间长短而定）的动物，或者可以挑选猫或狗（同时必须收养此类动物中“最老”的）。换言之，收养人不能自由挑选想收养的对象。请创建适用于这个系统的数据结构，实现各种操作方法，比如enqueue、dequeueAny、dequeueDog和dequeueCat。允许使用Java内置的LinkedList数据结构。

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

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

输入：
["AnimalShelf", "enqueue", "enqueue", "dequeueCat", "dequeueDog", "dequeueAny"]
[[], [[0, 0]], [[1, 0]], [], [], []]
 输出：
[null,null,null,[0,0],[-1,-1],[1,0]]

输入：
["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：最直观的想法是，使用一个队列存储cat，使用一个队列存储dog，其只存储动物编号。当动物种类为猫则压入cat，反之为狗则压入dog；当cat为空则删除dog，当dog为空则删除cat，反之按照cat和dog的动物编号大小来决定删除猫还是狗；当猫(狗)不为空则删除，反之返回-1。

AnimalShelf() {

    }

    queue<int> cat;
    queue<int> dog;
    
    void enqueue(vector<int> animal) {
        int id=animal[0];
        int type=animal[1];
        //猫
        if(type==0)
            cat.push(id);
        //狗
        else
            dog.push(id);
    }
    
    vector<int> dequeueAny() {
        //没有猫
        if(cat.empty())
            return dequeueDog();
        //没有狗
        if(dog.empty())
            return dequeueCat();
        //都不为空 按照id比对时间
        if(cat.front()<dog.front())
            return dequeueCat();
        else
            return dequeueDog();
    }
    
    vector<int> dequeueDog() {
        int id;
        if(!dog.empty())
        {
            id=dog.front();
            dog.pop();
            return {id,1};
        }
        return {-1,-1};
    }
    
    vector<int> dequeueCat() {
        int id;
        if(!cat.empty())
        {
            id=cat.front();
            cat.pop();
            return {id,0};
        }
        return {-1,-1};
    }

总结：一开始我在想，删除任何一种，难道我要新创建一个队列all嘛，那如果删除cat或者dog的话，我处理完cat或者dog后，如何处理all呢？后来我发现可以根据id来判断进来的时间早晚。