TreeSet: 不但可以去除重复元素,还可以对set集合中的元素进行排序, 底层数据结构是二叉树。简答归纳TreeSet保证元素的唯一性的依据:判断compareTo()方法是否return 0,如果返回0就认为是相同元素,就不会存储该元素。
下面创建一个TreeSet集合,集合中存放的是Intege类型的对象。
package com.test.list;
import java.util.Set;
import java.util.TreeSet;
public class TreeSetDemo
{
public static void main(String[] args)
{
Set<Integer> set = new TreeSet<Integer>();
set.add(3);
set.add(2);
set.add(3);
set.add(5);
set.add(9);
System.out.println("Size = "+ set.size());
System.out.println(set);
}
}
打印结果为:
Size = 4
[2, 3, 5, 9]
通过结果可以看出,集合中已将重复的元素去除,并且元素是按数字的大小顺序排列的。
下面在集合中存放一些自定义类型的元素:
package com.test.list;
import java.util.Set;
import java.util.TreeSet;
public class TreeSetDemo
{
public static void main(String[] args)
{
Set<Person> set = new TreeSet<Person>();
set.add(new Person("Linda", 33));
set.add(new Person("kevin", 27));
set.add(new Person("Joace", 25));
System.out.println(set);
}
}
class Person
{
private String name;
private int 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 Person(String name, int age)
{
this.name = name;
this.age = age;
}
}
运行后会抛出异常
Exception in thread "main" java.lang.ClassCastException: com.test.list.Person cannot be cast to java.lang.Comparable
at java.util.TreeMap.put(TreeMap.java:542)
at java.util.TreeSet.add(TreeSet.java:238)
at com.test.list.TreeSetDemo.main(TreeSetDemo.java:10)
之所以会出现该异常,是因为在创建类Person是没有指定比较算法,也就是集合中的元素不知道按照什么来比较大小,无法确定元素的存放位置。
使用TreeSet时必须指定比较的算法,指定的方式有两种:
自然顺序:要存储的类必须实现Comparable接口,并且复写compareTo()方法,在方法中指定算法。
比较器顺序:在创建TreeSet时,传入一个比较器Comparator,复写compare()方法,在该方法中指定算法。
//TreeSet排序的第一种方式:
//让元素自身具备比较性,元素需要实现Comparable接口,复写compareTo()方法,这种方式也称为自然顺序
package com.test.list;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;
public class TreeSetDemo
{
public static void main(String[] args)
{
Set<Person> set = new TreeSet<Person>();
set.add(new Person("Linda", 33));
set.add(new Person("kevin", 27));
set.add(new Person("Joace", 25));
set.add(new Person("Joace", 36));
for(Iterator<Person> it = set.iterator(); it.hasNext();)
{
Person person = it.next();
System.out.println(person.getName()+", "+person.getAge());
}
}
}
//要让类Person实现Comparable接口
class Person implements Comparable<Person>
{
private String name;
private int 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 Person(String name, int age)
{
this.name = name;
this.age = age;
}
//要复写compareTo()方法
public int compareTo(Person o)
{
//按年龄的大小来排序
int num = new Integer(this.age).compareTo(o.age);
//当主要条件相同时,必须要比较次要条件,否则,当年龄相等时,则认为是重复元素被去除
if(0 == num)
return new Integer((this.name).compareTo(o.name));
return num;
}
}
/*
排序的第二种方式:
当元素自身不具备比较性,或者具备的比较性不是所需要的,这是可以让集合自身具备比较性,需要自己定义一个比较器。
比较器:定义一个类,实现Comparator接口,复写compare方法。
当两种比较方式都存在时,以比较器为主。
* */
package com.test.list;
import java.util.Comparator;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;
public class TreeSetDemo
{
public static void main(String[] args)
{
Set<Person> set = new TreeSet<Person>(new MyCompare());
set.add(new Person("Linda", 33));
set.add(new Person("kevin", 27));
set.add(new Person("Joace", 33));
set.add(new Person("Joace", 36));
for(Iterator<Person> it = set.iterator(); it.hasNext();)
{
Person person = it.next();
System.out.println(person.getName()+", "+person.getAge());
}
}
}
//要让类Person实现Comparable接口
class Person implements Comparable<Person>
{
private String name;
private int 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 Person(String name, int age)
{
this.name = name;
this.age = age;
}
//要复写compareTo()方法
public int compareTo(Person o)
{
//按年龄的大小来排序
int num = new Integer(this.age).compareTo(o.age);
//当主要条件相同时,必须要比较次要条件,否则,当年龄相等时,则认为是重复元素被去除
if(0 == num)
return new Integer((this.name).compareTo(o.name));
return num;
}
}
class MyCompare implements Comparator<Person>
{ //复写compare()方法
public int compare(Person o1, Person o2)
{
//为了和自然顺序有区别,这里使用倒序
int num = new Integer(o2.getAge()).compareTo(new Integer(o1.getAge()));
return num != 0 ? num : o2.getName().compareTo(o1.getName());
}
}