题目描述
有家动物收容所只收留猫和狗,但有特殊的收养规则,收养人有两种收养方式,第一种为直接收养所有动物中最早进入收容所的,第二种为选择收养的动物类型(猫或狗),并收养该种动物中最早进入收容所的。
给定一个操作序列int[][2] ope(C++中为vector<vector<int>>)代表所有事件。若第一个元素为1,则代表有动物进入收容所,第二个元素为动物的编号,正数代表狗,负数代表猫;若第一个元素为2,则代表有人收养动物,第二个元素若为0,则采取第一种收养方式,若为1,则指定收养狗,若为-1则指定收养猫。请按顺序返回收养的序列。若出现不合法的操作,即没有可以符合领养要求的动物,则将这次领养操作忽略。
测试样例:
[[1,1],[1,-1],[2,0],[2,-1]]
返回:
[1,-1]
import java.util.*;
public class CatDogAsylum {
public ArrayList<Integer> asylum(int[][] ope) {
// write code here
ArrayList<Integer> res = new ArrayList<Integer>();
if(ope.length==0) return res;
LinkedList<Animal> dog = new LinkedList<Animal>();
LinkedList<Animal> cat = new LinkedList<Animal>();
int cnt = 0;
for(int i=0; i<ope.length; ++i){
int op = ope[i][0];
if(op==1){
if(ope[i][1] > 0){
dog.add(new Animal(ope[i][1], cnt++));
}else if(ope[i][1] < 0){
cat.add(new Animal(ope[i][1], cnt++));
}
}else if(op==2){
if(ope[i][1]==0){
if(!dog.isEmpty()&&!cat.isEmpty()){
if(dog.peek().order < cat.peek().order){
res.add(dog.peek().number);
dog.pop();
}else if(dog.peek().order > cat.peek().order){
res.add(cat.peek().number);
cat.pop();
}
}else if(!dog.isEmpty()){
res.add(dog.peek().number);
dog.pop();
}else{
res.add(cat.peek().number);
cat.pop();
}
}else if(ope[i][1]==1&&!dog.isEmpty()){
res.add(dog.peek().number);
dog.pop();
}else if(ope[i][1]==-1&&!cat.isEmpty()){
res.add(cat.peek().number);
cat.pop();
}
}
}
return res;
}
}
class Animal{
public int order;
public int number;
Animal(int number, int order){
this.number = number;
this.order = order;
}
}