您的位置:首页 > 其它

在TreeSet中实现存储自定义对象,并且能够按照特定的顺序排序

2015-04-20 17:29 573 查看
package cn.itcast.demo;

import java.util.*;
class Person implements Comparable
{
	private String name;
	private int age;
	private int score;
	
	Person(String name,int age,int score)
	{
		this.name=name;
		this.age=age;
		this.score=score;
	}
	public int getAge()
	{
		return age;
	}
	public String getName()
	{
		return name;
	}
	public int getScore()
	{
		return score;
	}
	public int compareTo(Object obj)
	{
		if (!(obj instanceof Person))
		{
			throw new RuntimeException("吊爆了!");
		}
		Person p=(Person)obj;
		if (this.age>p.age)
		{
			return 1;
		}
		if (this.age==p.age)
		{
			return this.name.compareTo(p.name);
		}
		return -1;
	}
}
class TreeSetTest1
{
	public static void main(String[] args)
	{
		//TreeSet t=new TreeSet(new Mycompare()); //按照姓名
		//TreeSet t=new TreeSet(new Mycompare1()); //按照年龄
		TreeSet t=new TreeSet(new Mycompare2()); //按照分数
		
		t.add(new Person("xxc1",15,99));
		t.add(new Person("xxc2",20,98));
		t.add(new Person("xxc6",17,108));
		t.add(new Person("xxc3",16,77));
		t.add(new Person("xxc5",10,230));
		Iterator t1=t.iterator();
		while (t1.hasNext())
		{
			Person p=(Person)t1.next();
			System.out.println(p.getName()+"======"+p.getAge()+"======"+p.getScore());
		}
	}
}
class Mycompare1 implements Comparator
{
	public int compare(Object o1,Object o2)
	{
		Person p=(Person)o1;
		Person p1=(Person)o2;
		//int num=p.getAge().compareTo(p1.getAge());
		int num = p.getAge() -p1.getAge();
		if (num==0)
		{
			return new Integer (p.getAge()).compareTo(new Integer(p1.getAge()));
		}
		return num;
	}
}

class Mycompare implements Comparator
{
	public int compare(Object o1,Object o2)
	{
		Person p=(Person)o1;
		Person p1=(Person)o2;
		int num=p.getName().compareTo(p1.getName());
		if (num==0)
		{
			return new Integer (p.getAge()).compareTo(new Integer(p1.getAge()));
		}
		return num;
		
	}
}

class Mycompare2 implements Comparator
{
	public int compare(Object o1,Object o2)
	{
		Person p=(Person)o1;
		Person p1=(Person)o2;
		//int num=p.getAge().compareTo(p1.getAge());
		int num = p.getScore() -p1.getScore();
		if (num==0)
		{
			return new Integer (p.getAge()).compareTo(new Integer(p1.getAge()));
		}
		return num;
	}
}

运行效果如下:



TreeSet的第一种排序方式是实现Comparable接口覆写compareTo方法,此排序是自然顺序排序。

TreeSet的第二种排序方式是实现Comparator接口覆写compare方法,此排序是在集合一初始化就拥有的,所以要讲此接口的实例对象传入到TreeSet的构造函数中。

2. TreeSet总结

1). TreeSet的特点

(1). 可以对元素进行排序

有两种排序方式。

(2). TreeSet保证元素的唯一性依据

在实现的Comparable的compareTo或者Comparator的compare方法中,如果这两个方法的返回值为0,那么TreeSet就认为这两个元素一样。按照Set的唯一性规则,在一次重复的元素不能被添加到TreeSet这个集合中。

2). TreeSet的两种排序方式

(1). 让元素本身具有比较性

元素本身要实现Comparable接口并实现里面的compareTo方法以保证元素本身具有比较性

(2). 让容器自身具有比较性

当元素本身不具有比较性或者具备的比较性不是所需要的,就在TreeSet建立实例的时候,传入Comparator接口的实现子类的实例。这个Comparator子类必须实现compare方法。

(3). 元素的可存储性

[1]. 自定义的类,如果重写了hashCode和equals两个方法,可以存入HashSet容器

[2]. 自定义的类,如果实现了Comparable的compareTo()方法,可以存入TreeSet容器。

【总结】如果自定义的类既重写了hashCode和equals,又实现了compareTo,那么这个类的元素既可以存入HashSet容器,也可以存入TreeSet容器。

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐