2020-03-30

Java基础【TreeSet】

字数统计: 514字 | 阅读时长: 2分

一、前言

我们都知道，TreeSet实现了SortedSet接口，会对集合元素自动排序。

那么，当我们需要存储自定义对象时，它是如何实现排序的呢？
在将数据存储到TreeSet中时，主要有两种方法实现对象的排序：
方法一：类必须实现Comparable接口

方法二：创建TreeSet对象时，指定Comparator接口的实现类对象（即：指定比较器）

二、Comparable接口

需求：有一学生类，该类有name和age属性，需要在TreeSet集合中存储该类的对象，并且希望先按照name属性升序排序，然后在考虑按照age属性排序。

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

    public Student() {
    }

    public Student(String name, int age) {
        this.name = name;
        this.age = age;
    }

    //setters and getters
    //toString()

    @Override
    public int compareTo(Student o) {
        int n1 = this.getName().compareTo(o.getName());
        int n2 = this.getAge()-o.getAge();
        return n1==0 ? n2:n1;
    }
}

public class Demo01 {
    public static void main(String[] args) {
        Set<Student> set = new TreeSet<>();
        set.add(new Student("xyz",24));
        set.add(new Student("aaa",22));
        set.add(new Student("hfw",21));
        set.add(new Student("aaa",21));
        System.out.println(set);
    }
}

测试结果如下：

1 2	[Student{name='aaa', age=21}, Student{name='aaa', age=22}, Student{name='hfw', age=21}, Student{name='xyz', age=24}]

三、Comparator接口

需求：存储不同的字符串，希望先按照字符串的长度升序排序，然后再按照字母进行排序。

public class Demo02 {
    public static void main(String[] args) {
        Set<String> set = new TreeSet<>(new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                int n1 = o1.length() - o2.length();
                int n2 = o1.compareTo(o2);
                return n1==0 ? n2:n1;
            }
        });

        set.add("helloworld");
        set.add("pingguo");
        set.add("lisi");
        set.add("zhangsan");
        set.add("beijing");
        set.add("cat");
        set.add("nanjing");
        set.add("xian");

        System.out.println(set);
    }
}

测试结果如下：

1	[cat, lisi, xian, beijing, nanjing, pingguo, zhangsan, helloworld]

四、总结

在存储自定义对象时，一定要记得要么实现Comparable接口，要么给TreeSet对象传递Comparator接口的实现类对象。
否则，程序会报错哦！

Java新手，若有错误，欢迎指正！

| 阅读量次