TreeSet存储自定义对象

本文详细介绍了如何使用TreeSet集合存储自定义对象,并确保对象的唯一性和排序。通过实现Comparable接口并重写compareTo方法,使得自定义的Person类能够被TreeSet正确排序和存储,避免ClassCastException异常。

TreeSet存储自定义对象

  • TreeSet集合是用来对象元素进行排序的,同样他也可以保证元素的唯一
    • 当compareTo方法返回0的时候集合中只有一个元素
    • 当compareTo方法返回正数的时候集合会怎么存就怎么取
    • 当compareTo方法返回负数的时候集合会倒序存储
package com.heima.bean;

public class Person implements Comparable<Person> {
	private String name;
	private int age;
	...
	...
	@Override
	public int compareTo(Person o) {
		return 1;
	}

}
package com.heima.set;

import java.util.TreeSet;

import com.heima.bean.Person;

public class Demo05_TreeSet {

	public static void main(String[] args)  {
		TreeSet<Person> ts = new TreeSet<>();
		ts.add(new Person("张三", 23));
		ts.add(new Person("李四", 13));
		ts.add(new Person("周七", 13));
		ts.add(new Person("王五", 43));
		ts.add(new Person("赵六", 33));
		
		System.out.println(ts);
	}
}

Exception in thread “main” java.lang.ClassCastException: com.heima.bean.Person cannot be cast to java.lang.Comparable
at java.util.TreeMap.compare(TreeMap.java:1188)
at java.util.TreeMap.put(TreeMap.java:531)
at java.util.TreeSet.add(TreeSet.java:255)
at com.heima.set.Demo05_TreeSet.main(Demo05_TreeSet.java:18)

如果自定义类不重写Comparable接口,就会报如上的错误。

查看Comparable的API,发现有很多类都实现了Comparable接口,也就是说有很多类都可以进行比较.
该接口中有一个compareTo方法,该方法可以让我们去完成比较.

怎么让自定义类可以进行比较呢,就去让该类去实现Comparable接口,并重写compareTo方法即可.

在Java中,TreeSet集合默认按照元素的自然顺序进行排序,但当存储自定义对象时,就需要为TreeSet指定排序规则。这通常是通过让自定义对象的类实现Comparable接口来完成的,或者通过提供一个Comparator来指定排序逻辑。以下是两种方式的示例: 1. 实现Comparable接口: 自定义类需要实现Comparable接口,并重写compareTo方法。这个方法定义了对象TreeSet中的排序规则。 ```java class Person implements Comparable<Person> { private String name; private int age; public Person(String name, int age) { this.name = name; this.age = age; } // Getter和Setter方法 @Override public int compareTo(Person other) { return Integer.compare(this.age, other.age); // 按照年龄排序 } } TreeSet<Person> set = new TreeSet<>(); set.add(new Person("Alice", 25)); set.add(new Person("Bob", 22)); ``` 在这个例子中,Person类实现了Comparable接口,并且根据年龄(age)进行排序。 2. 提供Comparator: 如果不希望或不能修改自定义类(比如使用第三方库中的类),则可以在创建TreeSet时提供一个Comparator。 ```java Comparator<Person> ageComparator = new Comparator<Person>() { @Override public int compare(Person p1, Person p2) { return Integer.compare(p1.getAge(), p2.getAge()); // 按照年龄排序 } }; TreeSet<Person> set = new TreeSet<>(ageComparator); set.add(new Person("Alice", 25)); set.add(new Person("Bob", 22)); ``` 在这个例子中,我们创建了一个Comparator来指定按年龄排序,并将其传递给TreeSet的构造函数。 两种方式都可以实现自定义对象TreeSet中的排序,选择哪一种取决于是否希望修改自定义对象的类,或者是否需要多个不同的排序规则。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

左绍骏

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值