Java集合中的排序方法

1.1.1. Comparable接口(了解)

缺省情况下,TreeSet中的元素会采用自然排序(从小到大),此时要求元素对象必须实现java.util.Comparable接口,大多数JDK自带的类都实现了该接口,比如八大包装类和String。

TreeSet会调用元素的compareTo方法来比较元素的大小关系,然后将集合元素按照升序排列。

public interface Comparable {

    public int compareTo(T o);

}

比较规则,拿当前元素和另一个元素做比较:

  • this > o:返回正整数 1 ,优先级较高

  • this < o:返回负整数 -1 ,优先级较低

  • this == o: 返回 0 ,此时认为两个对象为同一个对象。

此时compareTo方法返回0,则认为两个对象是同一个对象,返回正数排前面,返回负数排后面。

如果我们自定义一个类,需要存储到TreeSet中,此时我们需要让该类实现Comparable接口,并覆盖compareTo方法,在该方法编写比较规则。

需求:按照用户的年龄从小到大排序

class User implements java.lang.Comparable {

    private String name;

    private int age;

    public User(String name, int age) {

        this.name = name;

        this.age = age;

    }

    //比较规则

    public int compareTo(User o) {

        if (this.age > o.age) {

            return 1;

        } else if (this.age < o.age) {

            return -1;

        }

        return 0;

    }

    public String toString() {

        return "User [name=" + name + ", age=" + age + "]";

    }

}

public class ComparableDemo{

    public static void main(String[] args) {

        Set set = new TreeSet<>();

        set.add(new User("stef", 28));

        set.add(new User("will", 17));

        set.add(new User("allen", 15));

        set.add(new User("Lucy", 22));

        System.out.println(set);

    }

}

输出结果:

[User [name=allen, age=15], User [name=will, age=17], User [name=Lucy, age=22], User [name=stef, age=28]]

第一步:第一个节点进去,直接作为根节点,无需比较

image.png

第二步:第二个节点进去和第一个节点作比较,此时的this表示第二个节点,如果compareTo返回-1,则插入节点往左边走。

Java集合中的排序方法_第1张图片
image.png

第三步:第三个节点进去和第一个节点做比较,返回-1,再和第二个节点做比较。

Java集合中的排序方法_第2张图片
image.png

第四步:由于TreeSet是平衡二叉树,如果树不平衡,会对节点进行调整。

Java集合中的排序方法_第3张图片
image.png

第五步:第四个节点进去先和will比较,再和stef比较。

Java集合中的排序方法_第4张图片
image.png

1.1.2. Comparator接口(了解)

TreeSet除了默认支持自然排序外,还支持自定义排序,此时需要在构建TreeSet对象时传递java.util.Comparator接口的实现类对象,Comparator表示比较器,里面封装比较规则。

public interface Comparator {

    int compare(T o1, T o2);

}

此时compare方法返回0,则认为两个对象是同一个对象,返回正数排前面,返回负数排后面。

需求:根据用户的名字长度排序,如果名字相同按照年龄排序。

class User {

    private String name;

    private int age;

    public User(String name, int age) {

        this.name = name;

        this.age = age;

    }

    public int getAge() {

        return age;

    }

    public String getName() {

        return name;

    }

    public String toString() {

        return "User [name=" + name + ", age=" + age + "]";

    }

}

public class App {

    public static void main(String[] args) {

        Set set = new TreeSet<>(**new NameLengthComparator()**);

        set.add(new User("James", 30));

        set.add(new User("Bryant", 22));

        set.add(new User("Allen", 28));

        set.add(new User("Will", 17));

        System.out.println(set);//

    }

}

class NameLengthComparatorimplements java.util.Comparator {

    public int compare(User o1, User o2) {

        int ret = o1.getName().length() - o2.getName().length();

        if (ret > 0) {

            return 1;

        } else if (ret < 0) {

            return -1;

        } else {

            if (o1.getAge() > o2.getAge()) {

                return 1;

            } else if (o1.getAge() < o2.getAge()) {

                return -1;

            } else {

                return 0;

            }

        }

    }

}

输出结果:

[User [name=Will, age=17], User [name=Allen, age=28], User [name=James, age=30], User [name=Bryant, age=22]]

小结:HashSet做等值查询效率高,TreeSet做范围查询效率高,在开发中一般使用HashSet就可以了。

若要获得最好的学习效果,需要配合对应教学视频一起学习。需要完整教学视频,请参看https://ke.qq.com/course/272077。

你可能感兴趣的