题目
有家动物收容所只收留猫和狗,但有特殊的收养规则,收养人有两种收养方式,第一种为直接收养所有动物中最早进入收容所的,第二种为选择收养的动物类型(猫或狗),并收养该种动物中最早进入收容所的。
给定一个操作序列int[][2] ope(C++中为vector<vector>)代表所有事件。若第一个元素为1,则代表有动物进入收容所,第二个元素为动物的编号,整数代表狗,负数代表猫;若第一个元素为2,则代表有人收养动物,第二个元素若为0,则采取第一种收养方式,若为1,则指定收养狗,若为-1则指定收养猫。请按顺序返回收养的序列。若出现不合法的操作,即没有可以符合领养要求的动物,则将这次领养操作忽略。
测试样例:
[[1,1],[1,-1],[2,0],[2,-1]]
返回:[1,-1]
解法
使用两个队列来完成所有的操作,对于第一种操作,我们只需要返回队列中第一个元素即可;对于第二个操作,我们就将所有不符合要求的数据存入辅助队列中,一直找到第一个满足条件的元素,那么就删除,然后再将辅助队列所有元素又归还原队列.而我们所需要做的就是判断每个数据的第一个元素,根据不同的值选择不同的情况;跟军数据的第二个元素确定是猫还是狗。
class CatDogAsylum {
public:
vector<int> asylum(vector<vector<int> > ope) {
// write code here
vector<int> res;
if(ope.size()==0)
return res;
for(int i=0;i<ope.size();i++)
{
if(ope[i][0]==1)
{
q1.push(ope[i][1]);
}
else
{
if(ope[i][1]==0)
{
if(!q1.empty())
{
res.push_back(q1.front());
q1.pop();
}
}
else
{
while(!q1.empty())
{
if((ope[i][1]<0 && q1.front()<0)||(ope[i][1]>0 && q1.front()>0))
{
res.push_back(q1.front());
q1.pop();
break;
}
else
{
q2.push(q1.front());
q1.pop();
}
}
while(!q1.empty())
{
q2.push(q1.front());
q1.pop();
}
while(!q2.empty())
{
q1.push(q2.front());
q2.pop();
}
}
}
}
return res;
}
private:
queue<int> q1;
queue<int> q2;
};