Java_基础—TreeSet保证元素唯一和自然排序

TreeSet存储自定义对象时，必须使该对象实现Comparable接口，并重写该方法。

public class Person implements Comparable<Person>{

@Override
    public int compareTo(Person o) {    
        //return 0;      //返回0，集合中只有一个元素
        return 1;        //返回1，集合怎么存就怎么取
        //return -1;     //返回-1，集合会将存储的元素倒叙
    }
}

二叉树：小的存储在左边（负数），大的存储在右边（正数），相等就不存（0）

compareTo方法，在TreeSet集合如何存储元素取决于 compareTo方法的返回值

• 返回0，集合中只有一个元素
• 返回-1，集合会将存储的元素倒叙
• 返回1，集合怎么存就怎么取

自定义一个Person类，属性为name, age

package com.soar.bean;

public class Person implements Comparable<Person>{
    private String name;
    private int age;


    public Person() {
        super();    
    }
    public Person(String name, int age) {
        super();
        this.name = name;
        this.age = age;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
    public int getAge() {
        return age;
    }
    public void setAge(int age) {
        this.age = age;
    }

    public String toString() {
        return "Person [name=" + name + ", age=" + age + "]";
    }

    //Alt + Shift + S → Generate hashCode() and equals()方法
    /*
     * final int prime = 31;
     * 为什么是31？
     * ①31是个质数，质数是能被1和自己本身整除的数 
     * ②31这个数既不大也不小
     * ③31这个数好算，2的五次方-1，2向左移动5位
     */
    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + age;
        result = prime * result + ((name == null) ? 0 : name.hashCode());
        return result;
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj)                //调用的对象和传入的对象是同一个对象
            return true;                //直接返回true
        if (obj == null)                //传入的对象为null
            return false;               //返回false
        if (getClass() != obj.getClass())   //判断两个对象对应的字节码文件是否是同一个
            return false;                   //如果不是，直接返回false
        Person other = (Person) obj;        //向下转型
        if (age != other.age)           //调用对象的年龄不等于传入对象的年龄
            return false;               //返回false
        if (name == null) {             //调用对象的姓名为null
            if (other.name != null)     //传入对象的姓名不为null
                return false;           //返回flase
        } else if (!name.equals(other.name))    //调用的姓名不等于传入对象的姓名
            return false;                   //返回false
        return true;                //返回true
    }

    @Override
    public int compareTo(Person o) {    
        int num = this.age - o.age;         //年龄是比较的主要条件
        return num == 0 ? this.name.compareTo(o.name) : num;    //姓名是比较的次要条件
    }

}

定义一个TreeSet类

package com.soar.set;

import java.util.TreeSet;

import com.soar.bean.Person;

public class Demo3_TreeSet {
    /*
     *A:案例演示
     *  TreeSet存储Integer类型的元素并遍历
     *  TreeSet集合是用来对元素进行排序的，同样他也可以保证元素的唯一
     *  当compareTo方法返回0的时候，集合中只有一个一个元素
     *  当compareTo方法返回正数的时候，集合会怎么存就怎么取
     *  当compareTo方法返回负数的时候，集合会倒序存储
     */
    public static void main(String[] args) {
        //add();
        TreeSet<Person> ts = new TreeSet<>();
        ts.add(new Person("张三",23));
        ts.add(new Person("李四",24));
        ts.add(new Person("王五",25));
        ts.add(new Person("赵六",26));
        ts.add(new Person("周七",26));

        System.out.println(ts); 
    }

    private static void add() {
        TreeSet<Integer> ts = new TreeSet<>();
        ts.add(1);
        ts.add(2);
        ts.add(2);
        ts.add(3);
        ts.add(3);

        System.out.println(ts);
    }

}