关于TreeSet集合中的比较器问题

最新推荐文章于 2025-06-17 20:29:01 发布

原创最新推荐文章于 2025-06-17 20:29:01 发布 · 978 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#集合 #TreeSet #比较器

集合框架专栏收录该内容

11 篇文章

订阅专栏

本文探讨了在Java中使用TreeSet集合时遇到的问题，特别是当尝试存储自定义数据类型时遇到的异常。文章通过代码示例展示了如何存入字符串并成功运行，接着解释了在存入自定义Person类时导致异常的原因，并提供了两种解决方案，包括实现Comparable接口和创建专门的Comparator类。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.问题的提出

TreeSet集合的底层数据结构是二叉树，因此我们往集合中存入的对象就必须遵从这个数据结构，也就是存入集合的对象必须具有可比性。

比如我们TreeSet集合中存入int型或String的数据，输出没有问题，因为它们本身具有可比性。但如果我们存入自定义的类型，就很可能不具有可比性。就会出现问题

2.集合中存入字符串的代码示例

import java.util.Iterator;
import java.util.TreeSet;

public class Test {

	public static void main(String[] args) {
		
		TreeSet<String> ts=new TreeSet<String>();
		
		ts.add("abc");
		ts.add("cba");
		ts.add("nba");
		ts.add("rgb");
		
		Iterator<String> it=ts.iterator();
		
		while(it.hasNext()){
			
			System.out.println(it.next());
		}
	}

}

输出结果为：

abc
cba
nba
rgb

3.在集合中存入自定义的数据类型

首先定义Person类

public class Person {

	private String name;
	private int age;
	
	
	public Person(String name, int age) {
		super();
		this.name = name;
		this.age = age;
	}
	public Person() {
		super();
		
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public int getAge() {
		return age;
	}
	public void setAge(int age) {
		this.age = age;
	}
		
}

运行出现异常

Exception in thread "main" java.lang.ClassCastException: cn.hpu.bean.Person cannot be cast to java.lang.Comparable

原因及解决的两种方式

原因：存入的自定义数据不具有可比性

解决方式：

1.让存入元素本身具有可比性。可让自定义的元素实现Comparable接口，重写其中的comparaTo方法

2.让TreeSet集合本身具有比较功能。定义一个比较类，实现Comparator接口，重写其中的compare方法。将该类对象作为参数传递给TreeSet集合的构造函数中

解决方式一：

public class Person implements Comparable<Person>{

	private String name;
	private int age;
	
	
	public Person(String name, int age) {
		super();
		this.name = name;
		this.age = age;
	}
	public Person() {
		super();
		
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public int getAge() {
		return age;
	}
	public void setAge(int age) {
		this.age = age;
	}

	public int compareTo(Person p) {
		int temp=this.getAge()-p.getAge();
		
		return temp==0?this.getName().compareTo(p.getName()):temp;
	}	
}

解决方式二：

自定义比较器

public class ComparaByName implements Comparator<Person> {

	public static void main(String[] args) {
	

	}

	public int compare(Person p1, Person p2) {
		int temp=p1.getName().compareTo(p2.getName());
		return temp==0?p1.getAge()-p2.getAge():temp;//返回值为正数，表明前一个的name值大于后面一个的name值，TreeSet会自动将这两个元素互换位置
	}

}

传入TreeSet集合的构造函数中

import java.util.Iterator;
import java.util.TreeSet;

public class Test {

	public static void main(String[] args) {
		
		TreeSet<Person> ts=new TreeSet<Person>(new ComparaByName());
		
		ts.add(new Person("lisi0",22));
		ts.add(new Person("lisi2",20));
		ts.add(new Person("lisi5",28));
		ts.add(new Person("lisi9",22));
		
		
		Iterator<Person> it=ts.iterator();
		
		while(it.hasNext()){
			Person p=it.next();
			
			System.out.println(p.getName()+":"+p.getAge());
		}
		
	}

}