TreeSet特点:
注意:TreeSet满足Set的特点
1.不能重复存储同一个对象,数据储存是以“红黑树”数据结构存储。也是一个平衡二叉树
2.跟Set不同的是以排序方式输出数据
TreeSet的自然排序
步骤1:存储的类要实现Comparable接口
步骤2:重写hashcode()方法和equals()方法
示例说明
先创建一个R.java实现comparable比较器类
public class R implements Comparable<R>{
private int count;
public R(int count) {
this.count = count;
}
public int getCount() {
return count;
}
public void setCount(int count) {
this.count = count;
}
@Override
public String toString() {
return "R{" +
"count=" + count +
'}';
}
/**
* 比较两个对象是否相等:
* 1,hashcode是否相同
* 2.equals是否是true
* @return
*/
@Override
public int hashCode() {
return this.getCount();
// return 1;
}
@Override
public boolean equals(Object obj) {
if (obj == this){
return true;
}else if (obj == null){
return false;
}else if (obj instanceof R){
R otherobj = (R)obj;
if (otherobj.getCount() == ((R) obj).getCount()){
return true;
}
}
return false;
// return true;
}
/**
* 此方法的返回值决定大小
* 若返回值》0,则当前this比该object小
* 若返回值《0,则当前this比该object大
* @param o
* @return
*/
@Override
public int compareTo(R o) {
return this.getCount()>= o.getCount()?1:-1;
}
}
新建TreeSetComparable.java
public class TreeSetComparable {
public static void main(String[] args){
/**
* 创建一个TreeSet集合对象
* 类型要实现Comparable接口
*/
TreeSet<R> set = new TreeSet<>();
set.add(new R(456));
set.add(new R(26));
set.add(new R(156));
set.add(new R(356));
set.forEach((s)->System.out.println(s.toString()));
}
}
运行结果:
TreeSet的定制排序
定制排序定义
tip:这里用的R.java跟上面是同一个文件,这里就不重复放图了
注意:定制排序要实现Comparator接口,可以看出此接口 是函数接口,只有一个抽象方法
接口方法,返回值是int (1 前者 > 后者 /// -1 前者 < 后者 /// 0 前者 == 后者)
示例说明
新建:TreeSetMainTest.java
import demo01.R;
public class TreeSetMainTest {
public static <T> void main(String[] args) {
// TODO Auto-generated method stub
//定制排序
TreeSet<R> set = new TreeSet<>((a,b)->a.getCount()>b.getCount() ? -1:1);
set.add(new R(123));
set.add(new R(13));
set.add(new R(132));
//以lambda表达式写法
set.forEach((v)->System.out.println(v.toString()));
System.out.println("***************************");
//以匿名内部类写法
TreeSet<R> set2 = new TreeSet<>(new Comparator<R>() {
@Override
public int compare(R o1, R o2) {
// TODO Auto-generated method stub
if (o1.getCount() >= o2.getCount()) {
return 1;
}else if (o1.getCount() == o2.getCount()) {
return 0;
}else {
return -1;
}
}
});
set2.add(new R(123));
set2.add(new R(456));
set2.add(new R(789));
set2.add(new R(654));
set2.forEach((v)->System.out.println(v));
}
}
运行结果
总结
java类可以通过实现comparator接口自定义该类比较大小的方法,从而达到定制排序