您的位置:首页 > 编程语言

猫狗收容所

2018-04-13 20:53 141 查看
有家动物收容所只收留猫和狗,但有特殊的收养规则,收养人有两种收养方式,第一种为直接收养所有动物中最早进入收容所的,第二种为选择收养的动物类型(猫或狗),并收养该种动物中最早进入收容所的。       给定一个操作序列int[][2] ope(C++中为vector<vector<int>>)代表所有事件。若第一个元素为1,则代表有动物进入收容所,第二个元素为动物的编号,正数代表狗,负数代表猫;若第一个元素为2,则代表有人收养动物,第二个元素若为0,则采取第一种收养方式,若为1,则指定收养狗,若为-1则指定收养猫。请按顺序返回收养的序列。若出现不合法的操作,即没有可以符合领养要求的动物,则将这次领养操作忽略。测试样例:
[[1,1],[1,-1],[2,0],[2,-1]]
返回:[1,-1]
解析:不难,比较麻烦,需要使用else if过滤掉非法操作,不能直接else。先判断是动物是进还是出,即每个一维数组的第一位,若是进则正常add。若为出则需要判断几种情况,要么是指定出狗(编号>0)或猫(编号<0),要么就出最早进的。需要使用两个ArrayList,一个用来存动物进出,一个用来存返回结果。
代码:
    public ArrayList<Integer> asylum(int[][] ope) {
        ArrayList<Integer> ret = new ArrayList<>();
        ArrayList<Integer> alls = new ArrayList<>();
        for (int i = 0; i < ope.length; ++i) {
            if (ope[i][0] == 1) {
                alls.add(ope[i][1]);
            } else if (ope[i][0] == 2) {
                if (ope[i][1] == 0) {
                    if (!alls.isEmpty()) {
                        ret.add(alls.get(0));
                        alls.remove(0);
                    }
                } else if (ope[i][1] == 1) {
                    for (int j = 0; j < alls.size(); ++j) {
                        if (alls.get(j) > 0) {
                            ret.add(alls.get(j));
                            alls.remove(j);
                            break;
                        }
                    }
                } else if (ope[i][1] == -1) {
                    for (int j = 0; j < alls.size(); ++j) {
                        if (alls.get(j) < 0) {
                            ret.add(alls.get(j));
                            alls.remove(j);
                            break;
                        }
                    }
                }
            }
        }

        return ret;
    }
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  算法 编程题