Java TreeSet 添加失敗的解決

更新時間：2020年09月28日 10:38:48 作者：予悅君兮善窈窕

這篇文章主要介紹了Java TreeSet 添加失敗的解決，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

Java 中TreeSet 是Set的一個子類。

Set、List、Map區(qū)別

Set是一個無序、不允許重復(fù)的集合。

List（ArrayList、Vector等）是有序、可重復(fù)的。

Map（HashMap等）是鍵值對。

public static void demo() {
    TreeSet<Person> ts = new TreeSet<>();
    ts.add(new Person("張三", 23));
    ts.add(new Person("李四", 13));
    ts.add(new Person("周七", 13));
    ts.add(new Person("王五", 43));
    ts.add(new Person("趙六", 33));
    System.out.println(ts);
  }

此處會報異常，異常類型為java.lang.ClassCastException

這是因為Person類沒有實現(xiàn)Comparable 接口，并實現(xiàn)compareTo函數(shù)。

compareTo函數(shù)就是TreeSet用來判斷是否唯一的函數(shù)。

public class TreeEntity implements Comparable<Object> {
    @JsonProperty(value = "Name")
    private String name = "";

    /**
     * 名稱(字段名）
     * 
     */
    public final String getName() {
      return name;
    }

    public final void setName(String value) {
      name = value;
    }

    @JsonProperty(value = "Header")
    private String header = "";

    /**
     * 顯示的名稱（字段別名）
     */
    public final String getHeader() {
      if (header.isEmpty()) {
        header = name;
      }
      return header;
    }

    public final void setHeader(String value) {
      header = value;
    }

    @Setter(AccessLevel.PROTECTED)
    @JsonProperty(value = "Childrens")
    private TreeSet<TreeEntity> childrens;

    /**
     * 子節(jié)點集合
     * 
     */
    public final TreeSet<TreeEntity> getChildrens() {
      return childrens;
    }

    public final void setChildrens(TreeSet<TreeEntity> value) {
      childrens = value;
    }

    @Override
    public int compareTo(Object o) {
      TreeEntity te = (TreeEntity) o;
      if (te == null)
        return 1;
      return this.getName().compareTo(te.getName());
    }

    /**
     * 構(gòu)造函數(shù)
     */
    public TreeEntity() {
      childrens = new TreeSet<TreeEntity>();
    }
  }
  }
}

入上圖中的TreeEntity類重寫了compareTo函數(shù)，則是通過name屬性來判斷是否唯一。

在TreeSet.add()函數(shù)中，會觸發(fā)此compareTo函數(shù)，如果判斷不唯一，不會添加進去，但是代碼不會報異常。

compareTo函數(shù)返回值說明：

為什么返回0，只會存一個元素，返回-1會倒序存儲，返回1會怎么存就怎么取呢？原因在于TreeSet底層其實是一個二叉樹機構(gòu)，且每插入一個新元素(第一個除外)都會調(diào)用compareTo()方法去和上一個插入的元素作比較，并按二叉樹的結(jié)構(gòu)進行排列。

1. 如果將compareTo()返回值寫死為0，元素值每次比較，都認為是相同的元素，這時就不再向TreeSet中插入除第一個外的新元素。所以TreeSet中就只存在插入的第一個元素。

2. 如果將compareTo()返回值寫死為1，元素值每次比較，都認為新插入的元素比上一個元素大，于是二叉樹存儲時，會存在根的右側(cè)，讀取時就是正序排列的。

3. 如果將compareTo()返回值寫死為-1，元素值每次比較，都認為新插入的元素比上一個元素小，于是二叉樹存儲時，會存在根的左側(cè)，讀取時就是倒序序排列的。

補充知識：compareTo方法重寫錯誤導(dǎo)致TreeSet中無法添加對象

問題描述：

定義了一個實現(xiàn)Comparable接口的類R，包含一個int變量count。在測試類中添加了一個count為-3的對象后，便無法添加count為1的對象。但是可以添加count比-3小的和count為正數(shù)的R對象。

錯誤原因：

compareTo方法重寫錯誤。

代碼：

package test20160302;
import java.util.TreeSet;
class R implements Comparable<Object>{
  int count;
  public R(int count){
    this.count = count;
    System.out.println("count:"+count);
  }
  public String toString(){
    return "R[count:"+this.count+"]";
  }
  public boolean equals(Object obj){
    if(this == obj)return true;
    if(obj!=null && obj.getClass()==R.class){
      return this.count == ((R)obj).count;
    }else
      return false;
  }

  public int compareTo(Object obj){
    R r = (R)obj;
    System.out.println("用來比較的數(shù)："+this.count);
    System.out.println("被比較的數(shù)："+r.count);
    return this.count<r.count?-1:this.count>1?1:0;
  }
}

public class TreeSetTest3 {
  public static void main(String[] args) {
    TreeSet ts = new TreeSet();
    ts.add(new R(-3));
    ts.add(new R(-1));
    System.out.println(ts);
  }
}

輸出：