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

最新推荐文章于 2024-06-28 16:55:03 发布

华师刘德华

最新推荐文章于 2024-06-28 16:55:03 发布

阅读量1.3w

点赞数 1

分类专栏： Java

本文链接：https://blog.csdn.net/xuxiang668/article/details/90701599

版权

Java 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

最近项目中遇到一个问题，一个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;
            }
        });
    }