javaSE关于TreeSet的笔记（4）

package collection;

import java.util.TreeSet;

public class TreeSetTest04 {
    public static void main(String[] args) {
        Customer c1 = new Customer(32);
        Customer c2 = new Customer(20);
        Customer c3 = new Customer(30);
        Customer c4 = new Customer(25);

        TreeSet<Customer> Customers = new TreeSet<>();
        // 出现异常，因为Person类没有实现java.lang.Comparable接口
        Customers.add(c1);
        Customers.add(c2);
        Customers.add(c3);
        Customers.add(c4);

        // 遍历
        for (Customer p : Customers){
            System.out.println(p);
        }
    }
}

// 需要实现Comparable接口,equals方法可以不写
class Customer implements Comparable<Customer>{
    int age;

    public Customer(int age) {
        this.age = age;
    }

    // 在这个方法中编写比较的规则
    @Override
    public int compareTo(Customer customer) {
        /*int age1 = this.age;
        int age2 = customer.age;
        if (age1 == age2){
            return 0;
        }else if (age1 > age2){
            return 1;
        }else{
            return -1;
        }*/
        return this.age - customer.age;
    }

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