【牛客】猫狗收容所

2021-10-03 02:00 249阅读 0赞

## 题目描述 ##

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

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