TreeSet实现类
TreeSet:(有序)存入 Integer 类型的数据,不能重复且有升序的遍历。
(无序):但是它遍历的时候不按照我们输入的顺序遍历。
TreeSet底层是二叉树,并且使用内部比较器(Comparable)进行比较实现。
package zhai.set.TreeSet;
import java.util.TreeSet;
/*
TreeSet 使用
*/
public class TreeSetDemo1 {
public static void main(String[] args) {
TreeSetDemo1 ts = new TreeSetDemo1();
ts.tree();
ts.treeString();
}
public void tree(){
// 创建TreeSet对象
TreeSet<Integer> treeSet = new TreeSet<>();
treeSet.add(91);
treeSet.add(91);
treeSet.add(53);
treeSet.add(45);
treeSet.add(62);
treeSet.add(81);
System.out.println(treeSet.size());
System.out.println(treeSet);
}
public void treeString(){
// 创建TreeSet对象
TreeSet<String> ts1 = new TreeSet<>();
ts1.add("g");
ts1.add("g");
ts1.add("e");
ts1.add("a");
ts1.add("c");
ts1.add("b");
ts1.add("d");
ts1.add("f");
System.out.println(ts1.size()); // 我们放入了 8 条数据,其中有一条是重复的
System.out.println(ts1);
}
}
输出:
5
[45, 53, 62, 81, 91] // Integer类实现了Comparable接口使用ComparaTo()方法进行实现升序
7
[a, b, c, d, e, f, g] // String类实现了Comparable接口使用ComparaTo()方法进行实现排序
使用自定义的内部比较器
package zhai.set.TreeSet;
public class Student implements Comparable<Student>{
private int age;
private String name;
private String sex;
public Student(int age, String name, String sex) {
this.age = age;
this.name = name;
this.sex = sex;
}
public int getAge() {
if (age<0||age>120){
System.out.println("请勿输入非法年龄,请重启...");
}
return age;
}
public void setAge(int age) {
this.age = age;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getSex() {
return sex;
}
public void setSex(String sex) {
this.sex = sex;
}
@Override
public String toString() {
return "Student{" +
"age=" + age +
", name='" + name + '\'' +
", sex='" + sex + '\'' +
'}';
}
@Override
public int compareTo(Student o) {
return this.getAge() - o.getAge();
}
}
输出:
3
[Student{age=1, name='wang', sex='女'}, Student{age=11, name='lisi', sex='女'}, Student{age=12, name='zhangsan', sex='奥力给'}]
使用自定义的外部比较器:
测试类:
package zhai.set.TreeSet;
import java.util.Comparator;
public class Student2 {
private int age;
private String name;
public Student2(int age, String name) {
this.age = age;
this.name = name;
}
public int getAge() {
return age;
}
public void setAge(int age) {
this.age = age;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
@Override
public String toString() {
return "Student2{" +
"age=" + age +
", name='" + name + '\'' +
'}';
}
}
// 外部比较器的第一种的写法
//class A implements Comparator {
//
// @Override
// public int compare(Object o1, Object o2) {
// return 0;
// }
//}
package zhai.set.TreeSet;
import java.util.Comparator;
import java.util.TreeSet;
public class TreeSetDemo3 {
public static void main(String[] args) {
// 外部比较器的第二种写法
// 直接使用匿名内部类的写法进行创建Comparator(外部比较器)
Comparator<Student2> com = new Comparator<Student2>() { // 匿名创建外部比较器并且实现Comparator中的compare()方法进行比较
@Override
public int compare(Student2 o1, Student2 o2) {
return o1.getName().compareTo(o2.getName());
}
};
TreeSet<Student2> treeSet = new TreeSet<>(com);
treeSet.add(new Student2(12,"zhangsan"));
treeSet.add(new Student2(10,"lisi"));
treeSet.add(new Student2(80,"wang"));
treeSet.add(new Student2(12,"zhangsan"));
// for (Student2 s:treeSet) {
// System.out.println(treeSet);
// }
System.out.println(treeSet.size());
System.out.println(treeSet);
}
}
输出:
3
[Student2{age=10, name='lisi'}, Student2{age=80, name='wang'}, Student2{age=12, name='zhangsan'}]
比较器的总结:注意:使用TreeSet实现类时,一定要写一个比较器(一般只要不是自定义类型的底层都是默默认的内部比较器),因为它的底层是二叉树要使用比较器去比较,不实现比较器的话会报错。
内部比较器:类实现Comparable接口,重写compareTo()方法返回值进行比较。
外部比较器:自定义的类去实现Comparator接口,重写该接口中的compare()方法进行比较。(多态的方式)
TreeSet:底层是二叉树结构,默认排序是中序。