【程序员面试金典】 03.06. 动物收容所(双队列)

1.题目

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/animal-shelter-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

分别为猫和狗建立对应的队列，dequeueAny的时候若猫和狗均存在的时候，比较一下哪一个时间更久，选择出队。

class AnimalShelf
{
public:
    AnimalShelf()
    {

    }

    void enqueue(vector<int> animal)
    {
        if(animal[1] == 0)
            queue_cats.push(animal);
        else
            queue_dogs.push(animal);
    }

    vector<int> dequeueAny()
    {
        if(!queue_dogs.empty() && queue_cats.empty())
            return dequeueDog();
        else if(queue_dogs.empty() && !queue_cats.empty())
            return dequeueCat();
        else if(queue_dogs.empty() && queue_cats.empty())
            return {-1,-1};
        else
        {
            if(queue_dogs.front()[0] < queue_cats.front()[0])
                return dequeueDog();
            return dequeueCat();
        }
    }

    vector<int> dequeueDog()
    {
        if (queue_dogs.empty())
        {
            return vector<int> {-1, -1};
        }

        vector<int> dog = queue_dogs.front();
        queue_dogs.pop();
        return dog;

    }

    vector<int> dequeueCat()
    {
        if (queue_cats.empty())
        {
            return vector<int> {-1, -1};
        }

        vector<int> cat = queue_cats.front();
        queue_cats.pop();
        return cat;
    }
private:
    queue<vector<int>>queue_cats;
    queue<vector<int>>queue_dogs;
};

/**
 * Your AnimalShelf object will be instantiated and called as such:
 * AnimalShelf* obj = new AnimalShelf();
 * obj->enqueue(animal);
 * vector<int> param_2 = obj->dequeueAny();
 * vector<int> param_3 = obj->dequeueDog();
 * vector<int> param_4 = obj->dequeueCat();
 */