coding……
但行好事 莫问前程

Java8实现Map/List/Set排序

Java8之前如果想对一个集合排序,那么集合元素要么实现了Comparable接口,要么另外定义一个继承于Comparator的比较器并实现compare方法,使用起来是非常麻烦的。在我之前的文章中也介绍了一种Guava的排序实现方案,也是非常简单的,有兴趣的同学可以去了解一下Guava Ordering 。Java8之后可以使用java.util包的Comparator比较器,实现对集合的排序,使用起来非常简单。

首先定义一个实体类:

@Getter
@Setter
@AllArgsConstructor
@ToString
public class Student {
    private Long id;

    private String name;

    private Integer score;
}
  • List排序

使用Stream对List进行排序:

@Test
public void testSortWithStream(){

    /*使用Java8 Stream order*/
    List<Student> sortedList = studentList.stream().sorted(Comparator.comparing(Student::getScore).reversed()).collect(Collectors.toList());
    System.out.println(studentList);
    System.out.println(sortedList);

    /*使用Java8 Stream order按照score、name逆序排序*/
    List<Student> sortedList1 = studentList.stream().sorted(Comparator.comparing(Student::getScore).thenComparing(Student::getName).reversed()).collect(Collectors.toList());
    System.out.println(sortedList1);
}

使用list的sort方法排序:

@Test
public void testSort(){
    Comparator<Student> compareByScoreAndNameReverse = Comparator.comparing(Student::getScore).thenComparing(Student::getName).reversed();
    studentList.sort(compareByScoreAndNameReverse);
    System.out.println(studentList);
}

还有一种不推荐使用的方法,Collections.sort():

@Test
public void testSort1(){
    Comparator<Student> compareByScoreAndNameReverse = Comparator.comparing(Student::getScore).thenComparing(Student::getName).reversed();
    Collections.sort(studentList, compareByScoreAndNameReverse);
    System.out.println(studentList);
}
  • Map排序

根据key对Map排序:

@Test
public void SortByKeyTest() {
    Map<String, Integer> unsortMap = new HashMap<>();
    unsortMap.put("z", 10);
    unsortMap.put("b", 5);
    unsortMap.put("a", 6);
    unsortMap.put("c", 20);
    unsortMap.put("d", 1);
    unsortMap.put("e", 7);
    unsortMap.put("y", 8);
    unsortMap.put("n", 99);
    unsortMap.put("g", 50);
    unsortMap.put("m", 2);
    unsortMap.put("f", 9);

    System.out.println("Original...");
    System.out.println(unsortMap);

    // sort by keys, a,b,c..., and return a new LinkedHashMap
    // toMap() will returns HashMap by default, we need LinkedHashMap to keep the order.
    Map<String, Integer> result = unsortMap.entrySet().stream()
            .sorted(Map.Entry.comparingByKey())
            .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue,
                    (oldValue, newValue) -> oldValue, LinkedHashMap::new));

    //Alternative way to sort a Map by keys, and put it into the "result" map
    Map<String, Integer> result2 = new LinkedHashMap<>();
    unsortMap.entrySet().stream()
            .sorted(Map.Entry.comparingByKey())
            .forEachOrdered(x -> result2.put(x.getKey(), x.getValue()));

    System.out.println("Sorted...");
    System.out.println(result);
    System.out.println(result2);
}

根据value对Map进行排序:

@Test
public void sortByCommonValue() {
    Map<String, Integer> unsortMap = new HashMap<>();
    unsortMap.put("z", 10);
    unsortMap.put("b", 5);
    unsortMap.put("a", 6);
    unsortMap.put("c", 20);
    unsortMap.put("d", 1);
    unsortMap.put("e", 7);
    unsortMap.put("y", 8);
    unsortMap.put("n", 99);
    unsortMap.put("g", 50);
    unsortMap.put("m", 2);
    unsortMap.put("f", 9);

    System.out.println("Original...");
    System.out.println(unsortMap);

    //sort by values, and reserve it, 10,9,8,7,6...
    Map<String, Integer> result = unsortMap.entrySet().stream()
            .sorted(Map.Entry.comparingByValue(Comparator.reverseOrder()))
            .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue,
                    (oldValue, newValue) -> oldValue, LinkedHashMap::new));
    
    //Alternative way
    Map<String, Integer> result2 = new LinkedHashMap<>();
    unsortMap.entrySet().stream()
            .sorted(Map.Entry.<String, Integer>comparingByValue().reversed())
            .forEachOrdered(x -> result2.put(x.getKey(), x.getValue()));

    System.out.println("Sorted...");
    System.out.println(result);
    System.out.println(result2);
}

如果value是自定义对象,使用某个成员变量进行排序:

@Test
public void sortedByObjectValueExample(){
    List<Student> studentList = Lists.newArrayList();
    studentList.add(new Student(1001L, "zhuoli", 99));
    studentList.add(new Student(999L, "Alice", 87));
    studentList.add(new Student(1345L, "Michael", 90));
    studentList.add(new Student(1024L, "Jane", 99));

    //List -> Map
    Map<Long, Student> studentMap = studentList.stream().collect(Collectors.toMap(Student::getId, ele -> ele));

    //sort value by student score
    Map<Long, Student> result = studentMap.entrySet().stream()
            .sorted(Map.Entry.comparingByValue(Comparator.comparing(Student::getScore).reversed()))
            .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue,
                    (oldValue, newValue) -> oldValue, LinkedHashMap::new));

    System.out.println(result);
}
  • Set排序

@Test
public void setSortTest(){
    /*使用Java8 Stream order,结果转List*/
    List<Student> sortedList = studentSet.stream().sorted(Comparator.comparing(Student::getScore).reversed()).collect(Collectors.toList());
    System.out.println(sortedList);

    /*使用Java8 Stream order,使用LinkedHashSet保持顺序*/
    LinkedHashSet<Student> sortedSet1 = Sets.newLinkedHashSet();
    studentSet.stream().sorted(Comparator.comparing(Student::getScore).reversed()).forEachOrdered(sortedSet1::add);
    System.out.println(sortedSet1);
}

示例代码:码云 – 卓立 – Java8集合排序

 

赞(0) 打赏
Zhuoli's Blog » Java8实现Map/List/Set排序
分享到: 更多 (0)

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址