![TreeSet介绍]( "TreeSet介绍")
一、TreeSet原理：
1.TreeSet存储对象的时候, 可以排序, 但是需要指定排序的算法
2.Integer能排序(有默认顺序), String能排序(有默认顺序), 自定义的类存储的时候出现异常(没有顺序)
3.如果想把自定义类的对象存入TreeSet进行排序, 那么必须实现Comparable接口，或者实现一个比较器
在类上implements Comparable
重写compareTo()方法
在方法内定义比较算法, 根据大小关系, 返回正数负数或零
在使用TreeSet存储对象的时候, add()方法内部就会自动调用compareTo()方法进行比较, 根据比较结果使用二叉树形式进行存储
4.TreeSet依靠TreeMap来实现
二、TreeSet实现有序的2种方式
1）、元素实现Comparable接口：
实例：
实体类User（实现Comparable接口）：
|
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
47
48
49
50
|
|
package com.model.elgin.collections;
public class Userimplements Comparable<User>{
private int id;
private String name;
private String phone;
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getPhone() {
return phone;
}
public void setPhone(String phone) {
this.phone = phone;
}
public User(int id,String name,String
phone) {
super();
this.id = id;
this.name = name;
this.phone = phone;
}
@Override
public String toString() {
return "User [id=" + id +", name=" + name
+", phone=" + phone +"]";
}
@Override
public int compareTo(User user) {
if(this.id>user.getId()){
;
} else if(this.id==user.getId()){
;
}
;
}
}
|
测试类：
|
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
|
|
package com.model.elgin.collections;
import java.util.Set;
import java.util.TreeSet;
public class TestTreeSet {
public static void main(String[]
args) {
/*
* TreeSet有序、唯一。
* 实现有序的2种方式：
* 1.集合中的元素对象实现Comparable接口，TreeSet声明使用无参构造
* 2.声明一个比较器 Comparator ，TreeSet声明使用有参构造，参数传入比较器的一个实例对象
*
*/
Set<User> set=new TreeSet<User>();
User user1=,"jim",
"13232326565");
User user2=,"jack",
"13232326566");
User user3=,"lily",
"13232326567");
User user4=,"lucy",
"13232326568");
User user5=,"lucy",
"13232326568");
set.add(user1);
set.add(user3);
set.add(user4);
set.add(user2);
set.add(user5);
System.out.println(set);
}
}
|
运行之后发现结果有序（根据id由小到大）排列
2）、通过自定义比较器实现（User类不需要实现Comparable接口）：
比较器类： UserComparator
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
|
package com.model.elgin.collections;
import java.util.Comparator;
public class UserComparatorimplements Comparator<User>
{
@Override
public int compare(User user1, User user2) {
if(user1.getId()>user2.getId()){
;
}else if(user1.getId()==user2.getId()){
;
}
;
}
}
|
测试类：
|
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
|
|
package com.model.elgin.collections;
import java.util.Comparator;
import java.util.Set;
import java.util.TreeSet;
public class TestTreeSet {
public static void main(String[]
args) {
/*
* TreeSet有序、唯一。
* 实现有序的2种方式：
* 1.集合中的元素对象实现Comparable接口，TreeSet声明使用无参构造
* 2.声明一个比较器 Comparator ，TreeSet声明使用有参构造，参数传入比较器的一个实例对象
*
*/
Comparator<User> comparator=new UserComparator();
Set<User> set=new TreeSet<User>(comparator);
User user1=,"jim",
"13232326565");
User user2=,"jack",
"13232326566");
User user3=,"lily",
"13232326567");
User user4=,"lucy",
"13232326568");
User user5=,"lucy",
"13232326568");
set.add(user1);
set.add(user3);
set.add(user4);
set.add(user2);
set.add(user5);
System.out.println(set);
}
}
|
运行，同样可以得到上述有序的结果。
