牛客网 | 猫狗收容所

来源:互联网 发布:基础设施建设投资数据 编辑:程序博客网 时间:2024/04/30 03:04

题目描述

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

       给定一个操作序列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        int length = ope.length;        ArrayList<Integer> inlist = new ArrayList<>();        ArrayList<Integer> outlist = new ArrayList<>();        for (int i = 0; i < length; i++) {            if(ope[i][0]==1)//进收养所            {                inlist.add(ope[i][1]);            }            else if(ope[i][0]==2)//出收养所            {                if(inlist.isEmpty())                    continue;                if(ope[i][1]==0)//取出第一个动物                {                    outlist.add(inlist.get(0));                    inlist.remove(0);                }                else if(ope[i][1]==1)//取出狗                {                    for (int j = 0; j < inlist.size(); j++) {                        if(inlist.get(j)>0)                        {                            outlist.add(inlist.get(j));                            inlist.remove(j);                            break;                        }                    }                }                else if(ope[i][1]==-1)//取出猫                {                    for (int j = 0; j < inlist.size(); j++) {                        if(inlist.get(j)<0)                        {                            outlist.add(inlist.get(j));                            inlist.remove(j);                            break;                        }                    }                }            }        }        return outlist;    }}



0 0