treeSet集合中添加对象并指定排序

treeSet集合中添加对象并指定排序

//main方法
 public static void main(String[] args) {
        Set list = new TreeSet<>();
        list.add(new Dog(11, "小花"));
        list.add(new Dog(2, "小花"));
        list.add(new Dog(2, "小李"));
        list.add(new Dog(4, "小花"));
        //遍历集合
        Iterator it = list.iterator();
        while (it.hasNext()) {
            System.out.println(it.next());
        }
    }

//实体类，对id进行排序，相同的会去重
@Data
@AllArgsConstructor
@NoArgsConstructor
public class Dog implements Comparable<Dog> {
    private int id;
    private String name;
    @Override
    public int compareTo(Dog o) {
        if (!(o instanceof Dog))
            throw new RuntimeException("不是学生对象");
        //将传递过来的对象强转为学生对象
        Dog s = (Dog) o;
        System.out.println(this.name + "...comparableTo..." + s.name);
        if (this.id > s.id) {
            return 1;    //如果当前对象大于了传入对象的id，则返回一个正数，表示大于
        } else if (this.id == s.id) {
            return 0;    //如果当前对象等于传递对象的id，则返回一个0，表示等于
        } else {
            return -1;    //如果以上两种情况都不是，则返回一个负数，表示当前对象小于传递的对象
        }
    }
}

运行结果

• 如有不懂可私信或评论，作者看到立马回复