练习TreeSet的自定义类型的排序规则

今天药忘吃喽~ 2023-02-28 05:25 94阅读 0赞

继续探讨存入TreeSet集合中元素的排序规则：

假设有一个用户类，存储int age和String name。  
自定义比较规则：  
先按照年龄的从小到大排序。  
如果年龄相同，按照名字的从小到大排序。

class User implements Comparable<User> { 
        int age;
        String name;
    
        // 构造方法
        public User() { 
    
        }
        public User(int age, String name) { 
            this.age = age;
            this.name = name;
        }
    
        // 重写toString()，以便打印输出
        @Override
        public String toString() { 
            return age + "-" + name;
        }
    
        // 重写equals() 和 hashCode()，存入集合的元素都写上这些方法，养成好习惯
        @Override
        public boolean equals(Object o) { 
            if (this == o) return true;
            if (o == null || getClass() != o.getClass()) return false;
            User user = (User) o;
            return age == user.age && Objects.equals(name, user.name);
        }
    
        @Override
        public int hashCode() { 
            return Objects.hash(age, name);
        }
    
        @Override
        public int compareTo(User u) { 
            // 年龄相等时，按名字升序
            if (age == u.age) { 
                return name.compareTo(u.name);  // 按名字升序
            } else {   // 年龄不相等时
                return age - u.age;  // 按年龄升序
            }
        }
    }

测试代码：

import java.util.Objects;
    import java.util.*;
    
    public class Test08 { 
        public static void main(String[] args) { 
            // 创建TreeSet集合对象
            Set<User> users = new TreeSet<>();
    
            // 添加User元素
            users.add(new User(30, "zhangyi"));
            users.add(new User(25, "zhanger"));
            users.add(new User(20, "zhangsan"));
            users.add(new User(20, "liyi"));
            users.add(new User(25, "lier"));
            users.add(new User(30, "lisan"));
    
            // 遍历集合，查看是否排序
            for (User u : users) { 
                System.out.print(u + " ");  //20-liyi 20-zhangsan 25-lier 25-zhanger 30-lisan 30-zhangyi
            }
            System.out.println();
    
        }
    }