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

左手的ㄟ右手 2022-06-12 04:26 172阅读 0赞

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);
        }
    
    }