java TreeSet去重与排序入门

TreeSet用途

相比于HashSet,放入TreeSet的数据是有序的.那么这种有序是如何实现的呢?

TreeSet有序的原因

如下,插入的顺序是c d a,输出的时候是a c d,说明排序了.那么是如何排序的呢?

public class Test {
    public static void main(String[] args) {
        TreeSet<String> ts = new TreeSet<>();
        ts.add("c");// 在add方法时就排序了
        ts.add("d");// "d".compareTo("c")
        ts.add("a");
        System.out.println(ts);// [a, c, d]
    }
}

需要深入探究add方法,可以看出是调用了Map的put方法.然后value为固定值PRESENT.这是一种设计模式,叫做适配器设计模式. 以插入"d"为例,此时参数e的值为"d"

进入put方法

查看是否有Comparator
在本例中没有,所以要用Comparable 此处可以看出Comparator的优先级高于Comparable,实际上调用的是该接口下的compareTo方法.

自定义类排序

自定义的类要实现排序,就要实现Comparable接口或者实现Comparator.

Comparable决定TreeSet顺序

下面这个例子实现Comparable的compareTo接口,然后无论什么,都返回0. 此时set中就只能插入一个元素了.因为再怎么判断都是重复数据.

public class Test {
    public static void main(String[] args) {
        TreeSet<Person> ts = new TreeSet<>();
        ts.add(new Person(18));
        ts.add(new Person(28));
        ts.add(new Person(8));
        ts.add(new Person(25));
        ts.add(new Person(20));
        // 由于Set不能重复，所以return 0只能输出一个数据，重复的数据不予输入。
        System.out.println(ts);// [18]
    }
}

class Person implements Comparable{
    private  int age;
    public Person(int age){
        this.age=age;
    }

    @Override
    public String toString() {
        return ""+age;
    }

    @Override
    public int compareTo(Object o) {
        return 0;
    }
}

下面例子实现Comparable接口的compareTo方法,并依据年龄决定TreeSet中元素的顺序.

public class Test {
    public static void main(String[] args) {
        TreeSet<Person> ts = new TreeSet<>();
        ts.add(new Person(18, "lisi"));
        ts.add(new Person(28, "wangwu"));
        ts.add(new Person(8, "xiaobai"));
        ts.add(new Person(25, "zhaosi"));
        ts.add(new Person(20, "liuneng"));
        System.out.println(ts);// [wangwu 28, zhaosi 25, liuneng 20, lisi 18, xiaobai 8]
    }
}

class Person implements Comparable {
    private int age;
    private String name;

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


    @Override
    public String toString() {
        return name + " " + age;
    }

    //依据年龄决定插入顺序
    @Override
    public int compareTo(Object o) {
        if (!(o instanceof Person)) {
            throw new ClassCastException("类型不对");
        }
        Person p = (Person) o;
        return p.age - this.age;
    }
}

Comparator.决定TreeSet顺序

上面例子中,如果不想用年龄来排序,想用name作为排序依据,如何做?可以在Treeset的构造函数中传入 Comparator对象,如下

public class Test {
    public static void main(String[] args) {
        TreeSet<Person> ts = new TreeSet<>((o1, o2) -> o1.getName().compareTo(o2.getName()));
        ts.add(new Person(18, "lisi"));
        ts.add(new Person(28, "wangwu"));
        ts.add(new Person(8, "xiaobai"));
        ts.add(new Person(25, "zhaosi"));
        ts.add(new Person(20, "liuneng"));
        System.out.println(ts);//[lisi 18, liuneng 20, wangwu 28, xiaobai 8, zhaosi 25]
    }
}

class Person implements Comparable {
    private int age;
    private String name;

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

    public String getName() {
        return name;
    }

    @Override
    public String toString() {
        return name + " " + age;
    }

    //依据年龄决定插入顺序
    @Override
    public int compareTo(Object o) {
        if (!(o instanceof Person)) {
            throw new ClassCastException("类型不对");
        }
        Person p = (Person) o;
        return p.age - this.age;
    }
}

TreeSet如何去重

treeset在排序中包含了去重,因为重复的key不会被添加到map中.

总结

• Treeset可以去重,可以有序
• Treeset有序依赖于实现Comparable接口或者有比较器Comparator