关于TreeSet

 

java.lang.Object
   |_ java.util.AbstractCollection<E>
        |_ java.util.AbstractSet<E>
              |_ java.util.TreeSet<E>

TreeSet类声明如下：

public class TreeSet<E>
    extends AbstractSet<E>
    implements SortedSet<E>, Cloneable, java.io.Serializable

因为实现了SortedSet类，所以具有自然排序的功能。

TreeSet和HashSet相同的地方，就是集合里面不允许有重复的元素。自然排序情况下，一个TreeSet中只允许存放同一类型的多个元素，这里要求不是自定义的类。例如：

Set treeSet = new TreeSet();
   treeSet.add(new String("aaa"));
   treeSet.add(new String("aaa"));
   treeSet.add(new String("bbb"));
   treeSet.add(new String("ccc"));
   treeSet.add(new String("aaa"));
   System.out.println(treeSet);

结果输出为：

[aaa, bbb, ccc]

这时，treeSet.size()=3。而且，它是经过排序的输出。

如果有多个类的对象都加入到TreeSet集合中，就会发生异常。比如：

treeSet.add(new String("aaa"));
   treeSet.add(new Integer(100));
   System.out.println(treeSet);

发生异常：

Exception in thread "main" java.lang.ClassCastException: java.lang.String
at java.lang.Integer.compareTo(Integer.java:35)
at java.util.TreeMap.compare(TreeMap.java:1093)
at java.util.TreeMap.put(TreeMap.java:465)
at java.util.TreeSet.add(TreeSet.java:210)
at org.shirdrn.TreeSetTest.main(TreeSetTest.java:18)

而对于自定义的类，它的对象只能存放一个，而且实现类不需要实现Comparable接口。

但是，如果想要存放多个自定义的类的对象，不实现Comparable接口就会发生java.lang.ClassCastException异常。因此，想要能够进行客户化排序，必须实现比较器。

实现Comparable接口，就要实现compareTo()方法。而TreeSet又不存储相同的元素，这就要求自定义的类重写hashCode()和equals()方法：

class Person implements Comparable{
private String name;
private Integer age;
public Integer getAge() {
   return age;
}
public void setAge(Integer age) {
   this.age = age;
}
public String getName() {
   return name;
}
public void setName(String name) {
   this.name = name;
}

public boolean equals(Object o){
   if(this == o){
    return true;
   }
   if(! (o instanceof Person)){
    return false;
   }
   final Person other = (Person)o;
   if(this.name.equals(other.getName()) && this.age.equals(other.getAge())){
    return true;
   }
   else{
    return false;
   }
}

public int hashCode(){
   int result;
   result = (name == null?0:name.hashCode());
   result = 37*result + (age == null?0:age.hashCode());
   return result;
}

public int compareTo(Object o){
   Person other = (Person)o;
   if(this.name.compareTo(other.getName()) > 0){
    return 1;
   }
   if(this.name.compareTo(other.getName()) < 0){
    return -1;
   }
   if(this.getAge().intValue() > other.getAge().intValue()){
    return 1;
   }
   if(this.getAge().intValue() < other.getAge().intValue()){
    return -1;
   }
   return 0;
}

}

测试一下：

Set treeSet = new TreeSet();
   Person p1 = new Person();
   p1.setName("shirdrn");
   p1.setAge(new Integer(26));
   treeSet.add(p1);
   Person p2 = new Person();
   p2.setName("shirdrn");
   p2.setAge(new Integer(26));
   treeSet.add(p2);
   System.out.println(treeSet);

实例化了2个Person对象，实际上他们是同一个，因此只输出一个：

[org.shirdrn.Person@c29b5984]

如果将p2的name设置为p2.setName("Keller")，则输出两个：

[org.shirdrn.Person@5462263d, org.shirdrn.Person@c29b5984]

由于在Person类中实现类了compareTo()方法，输出结果是排序的，首先按照name排序，然后再按照age排序：

while(it.hasNext()){
    Person p = (Person)it.next();
    System.out.println("name = "+p.getName()+" || age = "+p.getAge());
   }

输出结果为：

name = Keller || age = 26
name = shirdrn || age = 26

name按照字母序排序。如果name相同，就按照age数字序排序。

TreeSet具有一些和HashSet类似的方法。

 

PS:如果不想去重写又想插入自定义的类

也可以用 public boolean addAll(Collection<? extends E> c) 方法！