TreeSet的另一种自定义排序：实现比较器接口

约定不等于承诺〃 2023-02-28 05:28 29阅读 0赞

#### TreeSet的另一种实现自定义排序：实现比较器接口 ####

1、使用TreeSet的有参数构造方法，传一个指定的比较器对象。

Comparable接口和Comparator接口怎么选择呢？到底用哪个好？

当比较规则不会发生改变的时候，或者说当比较规则只有1种的时候，建议实现Comparable接口。

比较规则有多种，并且需要多个规则之间来回切换，建议实现Comparator接口。

其中Comparator接口的设计符合OCP原则。

--------------------

假设有一个Customer类，待比较的对象：

class Customer { 
        int age;
        String name;
    
        public Customer() { 
    
        }
        public Customer(int age, String name) { 
            this.age = age;
            this.name = name;
        }
    
        @Override
        public String toString() { 
            return age + "-" + name;
        }
    }

--------------------

设计比较器对象，实现java.util.Comparator接口：

class CustomComparator implements Comparator<Customer> { 
        @Override
        public int compare(Customer o1, Customer o2) { 
            // 年龄相同按照名字升序
            if (o1.age == o2.age) { 
                return o1.name.compareTo(o2.name);
            } else {   // 年龄不相同
                return o1.age - o2.age;  // 按照年龄升序
            }
        }
    }

--------------------

测试代码：

import java.util.*;
    
    public class Test09 { 
        public static void main(String[] args) { 
            // 创建TreeSet对象
            //Set<Customer> customs = new TreeSet<>(new CustomComparator());
    
            // 也可以使用匿名内部类，直接new接口
            Set<Customer> customs = new TreeSet<>(new Comparator<Customer>() { 
                @Override
                public int compare(Customer o1, Customer o2) { 
                    // 年龄相同按照名字升序
                    if (o1.age == o2.age) { 
                        return o1.name.compareTo(o2.name);
                    } else {   // 年龄不相同
                        return o1.age - o2.age;  // 按照年龄升序
                    }
                }
            });
    
            // 添加Customer对象
            customs.add(new Customer(30, "zhangyi"));
            customs.add(new Customer(25, "zhanger"));
            customs.add(new Customer(20, "zhangsan"));
            customs.add(new Customer(20, "liyi"));
            customs.add(new Customer(25, "lier"));
            customs.add(new Customer(30, "lisan"));
    
            // 遍历集合
            for (Customer c : customs) { 
                System.out.print(c + " ");  //20-liyi 20-zhangsan 25-lier 25-zhanger 30-lisan 30-zhangyi
            }
            System.out.println();
    
            /* 所以存到TreeSet集合中的元素要想做到排序，有2种方法： 第一种：元素类实现java.lang.Comparable接口，重写比较规则。 第二种：在TreeSet的有参数构造方法中传一个比较器对象。 * */
        }
    }

--------------------