Java学习——TreeSet底层红黑树理解

最新推荐文章于 2024-07-12 10:19:39 发布

一个Debug都不会的废物

最新推荐文章于 2024-07-12 10:19:39 发布

阅读量1.1k

点赞数 1

文章标签：数据结构算法机器学习

本文链接：https://blog.csdn.net/python0731/article/details/120683210

版权

首先介绍下红黑树的红黑规则

每一个节点或是红色的,或者是黑色的
根节点必须是黑色
如果一个节点没有子节点或者父节点,则该节点相应的指针属性值为Nil,这些Nil视为叶节点,每个叶节点(Nil)是黑色的
如果某一个节点是红色,那么它的子节点必须是黑色(不能出现两个红色节点相连的情况)
对每一个节点,从该节点到其所有后代叶节点的简单路径上,均包含相同数目的黑色节点

总结下来,可归结为以下几个特征

每一个节点是红色或者黑色,根节点必须是黑色
每个叶节点(Nil)是黑色的
不能出现两个红色节点相连
对每一个节点,到其所有的后代节点的简单路径上,均包含相同的黑色节点

有人会问,既然根节点可以是红色,也可以是黑色,那么就设计到如何抉择的问题!!!!

首先默认节点都为黑色 ,添加第一个节点20

作为根节点满足红黑树的第一个规则

接下来添加第二个节点,18

18跟20相比比20小,所以存左边

那么此时,你可能发现了,现在不能买满足红黑树的第4个规则,即每一个节点,到其所有的后代节点的简单路径上,均包含相同的黑色节点(简单路径就是单向,不能回头),所以此时我们要将18改为红色

改为红色之后就满足红黑树所有规则了

最后我们插入23这个节点,比20大所以存放在右边

从根节点左子树到叶子节点 2个黑色节点

从根节点右子树到叶子节点 3个黑色节点,

又破坏了第4条规则,所以此时将23节点改为红色

所以默认节点都为黑色的时候,添加3个元素,一共需要调整两次!

下面我们默认将上面3个节点的颜色改为红色

再实现以上操作

插入第一个节点,不满足红黑树的第一个规则,根节点为黑色,其他都满足,所以将20节点改为黑色

接下去插入第二个节点18

完全满足红黑树的规则,再接下去插入第三个节点23,发现也完全满足红黑树的4个规则,所以我们发现,在默认节点都为红色的请款下,添加三个元素,只需要调节根节点那一次就够了!

所以对比以上两种方法,在节点默认为红色的时候,效率更高!

红黑树添加节点小结

以上总结,光看图解,很难理解,需要去配合视频一起食用,效果最好!

下面是进行红黑树练习的时候遇到的一些问题!!!!!

案例是:成绩排序

package com.NMSL.yyds;

import java.util.TreeSet;

public class TreeSet_Test {
    public static void main(String[] args) {
        TreeSet<Student> treeSet = new TreeSet<>();

        Student s1 = new Student("xiaoming",100,90,80);
        Student s2 = new Student("xiaohua",71,94,89);
        Student s3 = new Student("xiaohei",71,89,94);

        treeSet.add(s1);
        treeSet.add(s2);
        treeSet.add(s3);


        for (Student student : treeSet) {
            System.out.println(student);  //这边为什么会错 是因为 TreeSet函数需要自己编写排序！首选接口！
        }

    }
}

在用TreeSet的时候,忘记要打印TreeSet集合,需要自己定义排序规则!

TreeSet集合特点

不可以存储重复元素
没有索引(不能用for循环,只能用增强for和迭代器)
可以将元素按照规则进行排序

TreeSet指定排序规则一般有2种方法

1.自然排序Comparable

使用空参构造创建TreeSet集合
用TreeSet集合存储自定义对象，无参构造方法使用的是自然排序对元素进行排序的
自定义的Student类实现Comparable接口
自然排序，就是让元素所属的类实现Comparable接口，重写compareTo(T o)方法
重写接口中的compareTo方法
重写方法时，一定要注意排序规则必须按照要求的主要条件和次要条件来写

2.比较器排序Comparator

用TreeSet集合存储自定义对象，带参构造方法使用的是比较器排序对元素进行排序的
比较器排序，就是让集合构造方法接收Comparator的实现类对象，重写compare(T o1,T o2)方法
重写方法时，一定要注意排序规则必须按照要求的主要条件和次要条件来写

下面是利用方法1写好的类!(一般情况下我们选择方法1,只要在不触动底层代码的情况下,都选1,否则选2)


package com.NMSL.yyds;

public class Student implements Comparable<Student> {
    private String name;
    private int chinese;
    private int math;
    private int english;


    public Student() {
    }

    public Student(String name, int chinese, int math, int english) {
        this.name = name;
        this.chinese = chinese;
        this.math = math;
        this.english = english;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getChinese() {
        return chinese;
    }

    public void setChinese(int chinese) {
        this.chinese = chinese;
    }

    public int getMath() {
        return math;
    }

    public void setMath(int math) {
        this.math = math;
    }

    public int getEnglish() {
        return english;
    }

    public void setEnglish(int english) {
        this.english = english;
    }


    @Override
    public String toString() {
        return "Student{" +
                "name='" + name + '\'' +
                ", chinese=" + chinese +
                ", math=" + math +
                ", english=" + english +
                '}'+"总分为"+getSum();
    }

    public int getSum(){
        return chinese+math+english;
    }
    @Override
    public int compareTo(Student o) {
        int result = this.getSum()-o.getSum();//
        result=result==0?this.chinese-o.chinese:result;//当总成绩相同，比较语文
        result=result==0?this.math-o.math:result;//当总成绩相同，语文相同，比较数学！


        result=result==0?this.name.compareTo(o.name):result;//当都相同，比较字母！


        return result;
    }
}

这边打印的顺序是从小到大打印的,如果需要从大到小打印,只需要在类里面将this.getSum()-o.getSum()改为 o.getSum()-this.getSum()(this是当前要存进去的节点,o是已经存进去的节点)