Java中List按照自定义的顺序来排序

最近项目中遇到一个问题,一个List中每个元素都包含省份信息,而现在需要按照给定好的省份顺序对这个List进行排序。
省份的顺序如下:北京、天津、河北、山西、内蒙古、辽宁、吉林、黑龙江、上海、江苏、浙江、安徽、福建、江西、山东、河南、湖北、湖南、广东、广西、海南、重庆、四川、贵州、云南、西藏、陕西、甘肃、青海、宁夏、新疆(其余省份或地区排列在后面)。取省份前两个字符,如下:

final List<String> regulationOrder = Arrays.asList("北京", "天津", "河北", "山西", "内蒙", "辽宁", "吉林", "黑龙", "上海", "江苏", "浙江", "安徽", "福建", "江西", "山东", "河南", "湖北", "湖南", "广东", "广西", "海南", "重庆", "四川", "贵州", "云南", "西藏", "陕西", "甘肃", "青海", "宁夏", "新疆");

排序代码如下:

private void sortedByProvince(List<Map<String, Object>> enrollmentGroups) {

        //返回某个招生组最优先的省份(一个招生组可能有多个省份)
        Function<Map<String, Object>, String> getProvince = (oneGroup) -> {
            String province;
            String provinceTemp = (String) oneGroup.get("province");

            if (provinceTemp.contains(",")) {
                String[] provinces = provinceTemp.replaceAll("\\s*", "").split(",");
                province = provinces[0].substring(0, 2);

                for (int i = 0; i < provinces.length; i++) {
                    if (regulationOrder.indexOf(province) == -1) {
                        province = provinces[i].substring(0, 2);
                    }else if (regulationOrder.indexOf(province) > regulationOrder.indexOf(provinces[i].substring(0, 2))) {
                        province = provinces[i].substring(0, 2);
                    }
                }
            } else {
                province = provinceTemp.substring(0, 2);
            }

            return province;
        };

        Collections.sort(enrollmentGroups, new Comparator<Map<String, Object>>() {
            @Override
            public int compare(Map<String, Object> o1, Map<String, Object> o2) {
                String o1_province = getProvince.apply(o1); //o1中最优先的省份
                String o2_province = getProvince.apply(o2); //o2中最优先的省份

                int o1_index = regulationOrder.indexOf(o1_province);
                int o2_index = regulationOrder.indexOf(o2_province);

                if (o1_index == -1) {
                    return 1;
                }

                if (o2_index == -1) {
                    return -1;
                }

                return o1_index - o2_index;
            }
        });
    }
发布了6 篇原创文章 · 获赞 0 · 访问量 3075
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览