《程序员面试金典》--狗猫收容所

题目描述:
   有家动物收容所只收留猫和狗，但有特殊的收养规则，收养人有两种收养方式，第一种为直接收养所有动物中最早进入收容所的，第二种为选择收养的动物类型（猫或狗），并收养该种动物中最早进入收容所的。给定一个操作序列int[][2] ope(C++中为vector<vector<int>>)代表所有事件。若第一个元素为1，则代表有动物进入收容所，第二个元素为动物的编号，整数代表狗，负数代表猫；若第一个元素为2，则代表有人收养动物，第二个元素若为0，则采取第一种收养方式，若为1，则指定收养狗，若为-1则指定收养猫。请按顺序返回收养的序列。若出现不合法的操作，即没有可以符合领养要求的动物，则将这次领养操作忽略。
测试样例：[[1,1],[1,-1],[2,0],[2,-1]]

返回：[1,-1]

题目分析：

  根据先进先出的原则，自然想到了用队列实现，如果只维护一个队列，那么第一种方法实现起来比较简单，只需要取出队头的动物则可以，但是第二种方法就比较复杂，需要访问整个队列来找出第一个被访问的猫或者狗。

  因此我们可以选择维护两个队列来实现，一个队列存放放入的狗，一个队列存放放入的猫，对于第二种方法实现起来相当容易，我们只需要根据要选择的猫或者狗从相应的队列中取出便可以，但是第一种方法需要判断那个两个队列头部的是猫先进入收容所，还是狗先进入，这个时候需要一个标志，所以我们每次把动物放入队列的时候，同时将一个递增的序号放入队列，这个序号就是一个时间序列，根据这个序号便可以轻松实现第一种方法。

代码实现如下：

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
class CatDogAsylum {
public:
    vector<int> asylum(vector<vector<int> > ope) {
        // write code here
		queue<int> cat;
		queue<int> dog;
		vector<int> vec;
		int index=0;
		int size1=ope.size();
		for(int i=0;i<size1;i++)
		{
			int kind=ope[i][0];
			if(kind==1)
			{
				if(ope[i][1]>=0)
				{
					dog.push(index++);
					dog.push(ope[i][1]);
				}
				else
				{
					cat.push(index++);
					cat.push(ope[i][1]);
				}
			}
			else
			{
				if(ope[i][1]==0)
				{
					int min=0;
					if(cat.empty()&&!dog.empty())
						min=1;
					if(!cat.empty()&&dog.empty())
						min=-1;
					if(!cat.empty()&&!dog.empty())
						min=dog.front()>cat.front()?-1:1;
					if(min==-1)
					{
						cat.pop();
						vec.push_back(cat.front());
						cat.pop();
					}
					if(min==1)
					{
						dog.pop();
						vec.push_back(dog.front());
						dog.pop();
					}
				}
				else
				{
					if(ope[i][1]==1&&!dog.empty())
					{
						dog.pop();
						vec.push_back(dog.front());
						dog.pop();
					}
					if(ope[i][1]==-1&&!cat.empty())
					{
						cat.pop();
						vec.push_back(cat.front());
						cat.pop();
					}
				}

			}
		}
		return vec;
    }
};