JavaSE 集合类TreeSet存储自定义对象

文章目录

• 一、自动排序功能测试
• 二、对自定义类的自动排序

一、自动排序功能测试

public class TreeSetDemo {
	public static void main(String args[]) {
			TreeSet<Integer> myTreeSet = new TreeSet<>();
			myTreeSet.add(9) ;
			myTreeSet.add(6) ;
			myTreeSet.add(8) ;
			myTreeSet.add(7) ;
			System.out.println(myTreeSet);
	}
}

运行结果：



可见TreeSet可以对存入的元素进行自动排序。

二、对自定义类的自动排序

比如我们的自定义类为Person

class Person{
	private String name;
	private int age ;
	public Person(String name, int age) {
		super();
		this.name = name;
		this.age = age;
	}
	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 String toString() {
		return "[name=" + name + ", age=" + age + "]";
	}
}

放入TreeSet里试试：

public class TreeSetDemo {
	public static void main(String args[]) {
			TreeSet<Person> myTreeSet = new TreeSet<>();
			myTreeSet.add(new Person("唐三",20)) ;
			myTreeSet.add(new Person("唐三",20)) ;
			myTreeSet.add(new Person("小舞",19)) ;
			myTreeSet.add(new Person("奥斯卡",22)) ;
			System.out.println(myTreeSet);
	}
}

运行就报错了：



因为Person对象间相互无法比较，需要我们定义一个比较的规则，比如按照年龄大小排序。于是我们需要认识到Comparable接口，实现了Comparable的类的对象都可以按照一定规则相互比较。查看下api文档，我们看到熟悉的Integer与String等都实现了Comparable，所以它们间才可以相互比较。



Comparable中只有一个方法：



TreeSet底层是一棵二叉树，小的元素存左边，大的元素存右边，若有了相同元素就不存，在TreeSet集合如何存取元素取决于compareTo方法的返回值，返回负数存左边，返回正数存右边，返回0就不存。(需要注意第一个存的元素不论返回值是什么都会存入二叉树根节点位置，是从第二元素开始遵守以上规律)

我们来看看具体如何实现，修改Person类再次运行TreeSetDemo：

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 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 String toString() {
			return "[name=" + name + ", age=" + age + "]";
		}
	public int compareTo(Person o) {
		int num = this.age-o.age;//这里的o是根结点上的Person
		return num;
	}
}

运行结果：



当然仅按年龄排序存储的弊端是当名字不同而年龄相同时，靠后者不会存。可以优化代码当年龄相同时判断名字是否相同，这里不再赘述。

感兴趣的朋友还可以试验以下情况：

1. compareTo方法恒返回0的时候集合中只有一个元素
2. 恒返回正数的时候集合存取顺序一致
3. 恒返回负数的时候集合会按存入的倒序存储