[集合]TreeSet的自然排序与定制排序

最新推荐文章于 2022-05-05 23:24:25 发布

原创最新推荐文章于 2022-05-05 23:24:25 发布 · 314 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#java #开发语言

Java 同时被 2 个专栏收录

85 篇文章

订阅专栏

集合

12 篇文章

订阅专栏

这篇博客详细介绍了Java中TreeSet的使用，包括添加元素的要求、自然排序和定制排序的方法。在自然排序中，比较对象相同的标准是compareTo()返回0；在定制排序中，使用Comparator并根据compare()返回0判断。示例展示了如何按照姓名和年龄进行排序，并通过iterator遍历输出排序后的User对象。

public class TreeSetTest {
    /*
    1.向TreeSet中添加的数据，要求是相同类的对象
    2.两种排序方式：自然排序(实现Comparable接口)和定制排序(Comparator)

    3.自然排序中，比较两个对象是否相同的标准为：compareTo()返回0，不再是equals()
    4.定制排序中，比较两个对象是否相同的标准为：compare()返回0，不再是equals()
     */

    @Test
    public void test1(){
        TreeSet treeSet = new TreeSet();

        //不能添加不同类的对象
//        treeSet.add(123);
//        treeSet.add(456);
//        treeSet.add("AA");
//        treeSet.add(new User("Tom",12));

        //举例一：
//        treeSet.add(34);
//        treeSet.add(-34);
//        treeSet.add(43);
//        treeSet.add(11);
//        treeSet.add(8);

        //举例二：
        treeSet.add(new User("Tom",12));
        treeSet.add(new User("Jerry",32));
        treeSet.add(new User("Jim",2));
        treeSet.add(new User("Mike",65));
        treeSet.add(new User("Jack",33));
        treeSet.add(new User("Jack",69));

        Iterator iterator = treeSet.iterator();
        while (iterator.hasNext()){
            System.out.println(iterator.next());
        }
    }

    @Test
    public void test2(){
        Comparator com = new Comparator() {
            //按照年龄从小到大排列
            @Override
            public int compare(Object o1, Object o2) {
               if(o1 instanceof User && o2 instanceof User){
                   User u1 = (User) o1;
                   User u2 = (User) o2;
                   return Integer.compare(u1.getAge(), u2.getAge());
               }else {
                   throw new RuntimeException("输入的数据类型不匹配");
               }
            }
        };

        TreeSet set = new TreeSet(com);
        set.add(new User("Tom",12));
        set.add(new User("Jerry",32));
        set.add(new User("Jim",2));
        set.add(new User("Mike",65));
        set.add(new User("Jack",33));
        set.add(new User("Mary",33));
        set.add(new User("Jack",69));

        Iterator iterator = set.iterator();
        while (iterator.hasNext()){
            System.out.println(iterator.next());
        }
    }
}

自定义User类

public class User implements Comparable{
    private String name;
    private int age;

    public User() {
    }

    public User(String name, int age) {
        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;
    }

    @Override
    public String toString() {
        return "User{" +
                "name='" + name + '\'' +
                ", age=" + age +
                '}';
    }

    @Override
    public boolean equals(Object o) {
        System.out.println("User equals()....");
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        User user = (User) o;

        if (age != user.age) return false;
        return name != null ? name.equals(user.name) : user.name == null;
    }

    @Override
    public int hashCode() {
        int result = name != null ? name.hashCode() : 0;
        result = 31 * result + age;
        return result;
    }

    //按照姓名从大到小排列，年龄从小到大排列
    @Override
    public int compareTo(Object o) {
        if (o instanceof User){
            User user = (User) o;
//            return this.name.compareTo(user.name);
            int compare = -this.name.compareTo(user.name);
            if (compare != 0){
                return compare;
            }else {
                return Integer.compare(this.age,user.age);
            }
        }else {
            throw new RuntimeException("输入的类型不匹配");
        }
    }
}