Java进阶-实现比较器接口

放到 TreeSet 或者 TreeMap 集合 key 部分的元素要想做到排序,包括两种方式:

  1. 放在集合中的元素实现 java.lang.Comparable 接口
  2. 在构造 TreeSet 或者 TreeMap 集合的时候给它传一个比较器对象

Comparable 和 Comparator 怎么选择呢?

  • 当比较规则不会发生改变时,或者说当比较规则只有 1 个的时候,建议实现 Comparable 接口。

  • 如果比较规则有多个,并且需要多个比较规则之间频繁切换,建议使用 Comparator 接口

  • Comparator 接口的设计符合 OCP 原则(开闭原则)

本章使用比较器的方式对集合中的元素进行排序,并展示匿名内部类的方式

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import java.util.Comparator;
import java.util.TreeSet;

public class TreeSetTest05 {

public static void main(String[] args) {
// 创建 TreeSet 集合的时候,需要使用这个比较器
// TreeSet<WuGui> wuGuis = new TreeSet<>(); // 这样不行,没有通过构造方法传递一个比较器

// 给构造方法传递一个比较器
TreeSet<WuGui> wuGuis = new TreeSet<>(new WuGuiComparator());

wuGuis.add(new WuGui(100));
wuGuis.add(new WuGui(50));
wuGuis.add(new WuGui(200));

for (WuGui wuGui : wuGuis) {
System.out.println(wuGui);
// WuGui [age=50]
// WuGui [age=100]
// WuGui [age=200]
}
}

class WuGui {
int age;
public WuGui(int age) {
super();
this.age = age;
}

@Override
public String toString() {
return "WuGui [age=" + age + "]";
}
}

// 单独在这里编写一个比较器
// 比较器实现 java.util.Comparator 接口
//(Comparable 是 java.lang 包下的,Comparator 是 java.util 包下的)
class WuGuiComparator implements Comparator<WuGui> {
@Override
public int compare(WuGui o1, WuGui o2) {
return o1.age - o2.age;
}
}