二叉树的了解
通过查阅API我们得知TreeSet集合是基于TreeMap的实现,而TreeMap是基于二叉树(红黑树)结构,也就是说TreeSet集合的底层使用的二叉树(红黑树)结构。
树结构:它也是数据结构中的一种。在计算机领域中树结构指的是倒立的树。
树结构存储的数据,每个数据也需要节点来保存。
而TreeSet集合底层是二叉树的数据结构,什么是二叉树呢?
二叉树:每个节点的下面最多只能有2个子节点。
说明:最多表示一个节点下面可以有两个子节点或者一个子节点或者没有子节点。
在二叉树的根节点左侧的节点称为左子树,在根节点的右侧的节点称为右子树。
既然已经得知TreeSet集合底层是二叉树,那么二叉树是怎样存储数据的呢?是怎样保证存储的数据唯一并有序的呢?
二叉树的存储流程:
当存储一个元素的时候,如果是树的第一个元素,这个元素就作为根节点。
如果不是第一个元素,那么就拿要存储的元素与根节点进行比较大小:
大于根元素:就将要存储的元素放到根节点的右侧,作为右叶子节点。
等于根元素:丢弃。
小于根元素:就将要存储的元素放到根节点的左侧,作为左叶子节点。
总结:二叉树是通过比较大小来保证元素唯一和排序的。
案例 一、
使用TreeSet存储Employee对象,比较两个属性
int age, int weight 先按照年龄进行升序排,如果年龄相等的话,按照体重升序排
import java.util.Set;
import java.util.TreeSet;
class Employee implements Comparable<Employee>{
String name;
int age;
int weight;
public Employee(String name, int age, int weight) {
this.name = name;
this.age = age;
this.weight = weight;
}
@Override
public String toString() {
return "Employee{" +
"name='" + name + '\'' +
", age=" + age +
", weight=" + weight +
'}';
}
@Override
public int compareTo(Employee o) {
//先按照年两比,如果年龄相等 就比较体重
int num = this.age - o.age;
if (num == 0) {
int num1 = o.weight - this.weight;
return num1;
}
return num;
}
}
public class Demo2 {
public static void main(String[] args) {
Set<Employee> set = new TreeSet<>();
set.add(new Employee("张三", 35, 78));
set.add(new Employee("李四", 26, 70));
set.add(new Employee("王五", 35, 72));
set.add(new Employee("赵六", 35, 79));
set.add(new Employee("陈七", 32, 59));
set.add(new Employee("高八", 32, 59));
System.out.println(set);
}
}
总结:使用TreeSet的时候需要类实现一个接口 Comparable这个接口去做比较排序