牛客网 在线编程 猫狗收容所

题目描述

        有家动物收容所只收留猫和狗，但有特殊的收养规则，收养人有两种收养方式，第一种为直接收养所有动物中最早进入收容所的，第二种为选择收养的动物类型（猫或狗），并收养该种动物中最早进入收容所的。

       给定一个操作序列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 static ArrayList<Integer> asylum(int[][] ope) {
		Queue<Integer[]> cat = new LinkedList<>();
		Queue<Integer[]> dog = new LinkedList<>();
		ArrayList<Integer> asylum = new ArrayList<>();
		for (int i = 0; i < ope.length; i++) {
			if (ope[i][0] == 1) {
				if (ope[i][1] > 0)
					dog.offer(new Integer[] { i, ope[i][1] });
				if (ope[i][1] < 0)
					cat.offer(new Integer[] { i, ope[i][1] });
			} else if (ope[i][0] == 2) {
				if (ope[i][1] == 0) {
					if (cat.isEmpty()) {
						asylum.add(dog.poll()[1]);
					} else if (dog.isEmpty()) {
						asylum.add(cat.poll()[1]);
					} else if (dog.peek()[0] > cat.peek()[0]) {
						asylum.add(cat.poll()[1]);
					} else if (dog.peek()[0] < cat.peek()[0]) {
						asylum.add(dog.poll()[1]);
					}
				} else if (ope[i][1] == 1 && !dog.isEmpty()) {
					asylum.add(dog.poll()[1]);
				} else if (ope[i][1] == -1 && !cat.isEmpty()) {
					asylum.add(cat.poll()[1]);
				}
				continue;
			}
		}
		return asylum;
	}
}