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