Java TreeSet類的簡(jiǎn)單理解和使用

更新時(shí)間：2020年02月05日 12:17:06 作者：西紅柿里沒(méi)有番茄

這篇文章主要介紹了Java TreeSet類的簡(jiǎn)單理解和使用,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

TreeSet類是Set接口的一個(gè)實(shí)現(xiàn)類，主要作用是用于對(duì)對(duì)象的排序以及確定存入對(duì)象的唯一性。給對(duì)象排序的方式有很多，比如一些基本類型int、String等類型就已經(jīng)提供了很多排序的方法了，但是這并不說(shuō)明TreeSet類就沒(méi)有什么用了。在一些時(shí)候我們需要自定義一些類，同時(shí)需要對(duì)這個(gè)類的對(duì)象進(jìn)行排序，那么這個(gè)時(shí)候我們就可以通過(guò)這個(gè)TreeSet類去自定義一個(gè)排序的條件。

現(xiàn)在通過(guò)一個(gè)簡(jiǎn)單的案例實(shí)現(xiàn)來(lái)實(shí)現(xiàn)這個(gè)排序和唯一性

首先自定義一個(gè)Student類，類里面有三個(gè)屬性，分別是姓名、學(xué)號(hào)、年齡

public class Student{
 
 
  private String name;
  private String num;
  private int age;
 
 
  Student(String name, String num, int age) {
    this.name = name;
    this.num = num;
    this.age = age;
  }
 
  public String getName() {
    return name;
  }
 
  public void setName(String name) {
    this.name = name;
  }
 
  public String getNum() {
    return num;
  }
 
  public void setNum(String num) {
    this.num = num;
  }
 
  public int getAge() {
    return age;
  }
 
  public void setAge(int age) {
    this.age = age;
  }
}

然后我們給這個(gè)類的三個(gè)屬性添加值，并且將類的實(shí)例化對(duì)象添加到TreeSet類中。

注意：TreeSet類是沒(méi)有g(shù)et方法的，要輸出里面的內(nèi)容得通過(guò)foreach循環(huán)或者迭代器輸出，如果直接通過(guò)print進(jìn)行輸出，顯示的是內(nèi)存地址對(duì)象。

然后我們來(lái)看下輸出結(jié)果

import java.util.Iterator;
import java.util.Set;

public class TreeSet {

  public static void main(String[] args) {


    Student s1 = new Student("張珊", "111213", 18);
    Student s2 = new Student("隔壁", "111215", 19);
    Student s3 = new Student("翠花", "111214", 12);
    Student s4 = new Student("老王", "111212", 11);
    Student s5 = new Student("老黑", "111212", 11);


    Set set = new java.util.TreeSet();

    set.add(s1);
    set.add(s2);
    set.add(s3);
    set.add(s4);
    set.add(s5);

//   通過(guò)迭代器輸出結(jié)果
    Iterator iterator = set.iterator();
    while (iterator.hasNext()) {
      Student o = (Student) iterator.next();
      System.out.println(o.getName());

    }


  }


}

可以看到無(wú)法正常輸出，報(bào)錯(cuò)。這里報(bào)錯(cuò)的原因是類型轉(zhuǎn)換錯(cuò)誤。因?yàn)閷?duì)象存入TreeSet類中時(shí)需要對(duì)對(duì)象進(jìn)行一個(gè)比較，第一次存入時(shí)因?yàn)闆](méi)有可對(duì)比的對(duì)象，所以不會(huì)報(bào)錯(cuò)，但是當(dāng)?shù)诙€(gè)對(duì)象存入時(shí)需要同第一個(gè)對(duì)象進(jìn)行比較，再?zèng)Q定在二叉樹(shù)中存放的位置。這里的比較方法需要我們自己去實(shí)現(xiàn)Comparable<>接口重寫一個(gè)compareTo()方法。

那么現(xiàn)在在自定義Student類中實(shí)現(xiàn)Comparable并且重現(xiàn)compareTo()方法

public class Student implements Comparable<Student>{
 
 
  private String name;
  private String num;
  private int age;
 
 
  Student(String name, String num, int age) {
    this.name = name;
    this.num = num;
    this.age = age;
  }
 
  public String getName() {
    return name;
  }
 
  public void setName(String name) {
    this.name = name;
  }
 
  public String getNum() {
    return num;
  }
 
  public void setNum(String num) {
    this.num = num;
  }
 
  public int getAge() {
    return age;
  }
 
  public void setAge(int age) {
    this.age = age;
  }
 
 
 
  @Override
  public int compareTo(Student o) {
//通過(guò)年齡的比較確定存放順序
    return this.age-o.age;
  }
}

現(xiàn)在看下輸出結(jié)果

可以看到存入的內(nèi)容可以正常輸出的了，但是老黑沒(méi)有輸出，這是因?yàn)槔虾诤屠贤醯哪挲g相同，而我們重寫的compareTo()方法中的比較條件就是年齡。下面來(lái)看看這個(gè)方法中的返回值的含義。

@Override
  public int compareTo(Student o) {
 
    return 0；<br data-filtered="filtered">　　　　　return 1;<br data-filtered="filtered">　　　　　return -1;<br data-filtered="filtered">
  }

compareTo()方法的返回值類型是int類型，在這里只存在三種情況，分別是：大于0，小于0，和等于0。

TreeSet的底層結(jié)構(gòu)是一個(gè)二叉樹(shù)，每次插入的對(duì)象都會(huì)根據(jù)二叉樹(shù)的結(jié)構(gòu)進(jìn)行排列。當(dāng)前后兩個(gè)對(duì)象的條件進(jìn)行比較返回正數(shù)時(shí)，后一個(gè)對(duì)象會(huì)存在已前一個(gè)對(duì)象為根的右節(jié)點(diǎn)；當(dāng)前后兩個(gè)對(duì)象的條件進(jìn)行比較返回負(fù)數(shù)時(shí)，后一個(gè)對(duì)象會(huì)存在已前一個(gè)對(duì)象為根的左節(jié)點(diǎn)；當(dāng)前后兩個(gè)對(duì)象的條件進(jìn)行比較返回0時(shí)，后一個(gè)對(duì)象不存入TreeSet中。因?yàn)槔虾诤屠贤跄挲g相同，所以這里名字為老黑的對(duì)象就不存入了。

存入TreeSet中的對(duì)象輸出的順序是按照二叉樹(shù)的中序進(jìn)行輸出的。

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: