Java 實現彙總排序

小碼code發表於2022-03-28
Java排序

排序在系統中經常能用到，一般可以在資料庫做排序，也可以在服務端做排序。在資料庫一般使用 order by 排序。而服務端也是使用快排。本期使用匯總排序。

問題

統計銷售資料，每個銷售員都有對應的部門和銷售量，現在要統計銷售資料。

要求部門總銷量遞減排序，相同部門的也按照遞減排序。

比如：

銷售員 部門 銷售額
A 南部 100w
B 南部 20W
C 北部 30W
D 北部 70W
E 北部 40W
F 東部 150W

根據彙總排序:

部門 銷售額
東部 150W
北部 130W
南部 120W

然後根據先按照部門總和排序，相同部門按照遞減排序:

銷售員 部門 銷售額
F 東部 150W
D 北部 70W
E 北部 40W
C 北部 30W
E 南部 100W
F 南部 20W

解決方案

前期建立 model

public class SalesmanStatistic {


    public SalesmanStatistic(String salesman, String department, Integer amount) {
        this.salesman = salesman;
        this.department = department;
        this.amount = amount;
    }

    @Override
    public String toString() {
        return "{salesman='" + salesman + '\'' +
                ", department='" + department + '\'' +
                ", amount=" + amount +
                '}';
    }

    private String salesman;

    private String department;

    private Integer amount;
}

新增資料

        // 填充資料
        List<SalesmanStatistic> list = new ArrayList<>();
        SalesmanStatistic statistic1 = new SalesmanStatistic("A","南方",100);
        SalesmanStatistic statistic2 = new SalesmanStatistic("B","南方",20);
        SalesmanStatistic statistic3 = new SalesmanStatistic("C","北方",30);
        SalesmanStatistic statistic4 = new SalesmanStatistic("D","北方",70);
        SalesmanStatistic statistic5 = new SalesmanStatistic("E","北方",40);
        SalesmanStatistic statistic6 = new SalesmanStatistic("F","東方",150);
        list.add(statistic1);
        list.add(statistic2);
        list.add(statistic3);
        list.add(statistic4);
        list.add(statistic5);
        list.add(statistic6);

使用兩個 map，key 都是存部門，第一個 map 的 value 銷售額的總額，第二個是存同部門的列表。

        Map<String,Integer> sumMap = new HashMap<>();
        Map<String, List<SalesmanStatistic>> listMap = new HashMap<>();
        // 把資料放在 map 裡面
        for (SalesmanStatistic statistic : list) {
            String key = statistic.getDepartment();
            Integer amount = statistic.getAmount();
            sumMap.put(key,sumMap.getOrDefault(key,0) + amount);
            List<SalesmanStatistic> subList = listMap.get(key);
            if (subList == null) {
                subList = new ArrayList<>();
            }
            subList.add(statistic);
            listMap.put(key,subList);
        }

首先進行總量 sumMap 排序,就是對 sumMap 的 value 進行排序：

        // 對總和 sumMap 排序
        List<Map.Entry<String,Integer>> sumMapList = new ArrayList<>(sumMap.entrySet());
        sumMapList.sort((o1, o2) -> o2.getValue().compareTo(o1.getValue()));

以上獲取到總和 sumMap 的列表，從大到小排列，然後遍歷每個資料，根據 key 匹配到 listMap 的 key。首先獲取對 listMap 的 value 列表進行排序，然後把 list 新增到總的集合裡面。

        List<SalesmanStatistic> list = new ArrayList<>();
        for (Map.Entry<String,Integer> entry : sumMapList) {
            List<SalesmanStatistic> list1 =  listMap.get(entry.getKey());
            list1.sort((o1, o2) -> o2.getAmount().compareTo(o1.getAmount()));
            list.addAll(list1);
        }
        list.stream().forEach(list3 -> System.out.println(list3.toString()));

列印輸出結果：

{salesman='F', department='東方', amount=150}
{salesman='D', department='北方', amount=70}
{salesman='E', department='北方', amount=40}
{salesman='C', department='北方', amount=30}
{salesman='A', department='南方', amount=100}
{salesman='B', department='南方', amount=20}

總結

根據部門彙總和進行排序，然後每個部門也按照從大到小排序。這裡使用到 map 的鍵值對屬性。流程如下：

  • 使用 sumMap 儲存部門總數以及使用 listMap 儲存部門資訊。
  • 對 sumMap 的 value 排序，把 map.entrySet 放在一個集合做排序。
  • 根據排序後的 sumMap 的 key 找到 listMap 的 value。先對列表排序，最後放在集合中。

如果覺得文章對你有幫助的話，請點個推薦吧!

相關文章