程式師世界 >> 編程語言 >> JAVA編程 >> JAVA綜合教程 >> TreeSet入門詳解，treeset入門

TreeSet入門詳解，treeset入門

編輯：JAVA綜合教程

TreeSet入門詳解，treeset入門

看到TreeSet先從最基礎的去了解他，他是一個類，他所在的位置是java.util包中。

我們可以看一看他的繼承架構圖：

該類實現的接口：

1TreeSet案例：

TreeSet是一個有序集合，TreeSet中的元素將按照升序排列，缺省是按照自然排序進行排列，意味著TreeSet中的元素要實現Comparable接口。或者有一個自定義的比較器。
我們可以在構造TreeSet對象時，傳遞實現Comparator接口的比較器對象。

import java.util.Iterator;
import java.util.*;

public class TreeSetTest {
    public static void main(String[] args) {
        Set ts = new TreeSet();
        ts.add("abc");
        ts.add("xyz");
        ts.add("rst");
        Iterator it = ts.iterator();
        while (it.hasNext()) {
            System.out.println(it.next());
        }

輸出：

abc

rst

xyz

打印結果不是和先前加入的順序一樣，它是按照一個字母的排序法進行排序的。這是因為String 類實現了Comparable接口。

如果我們自己定義的一個類的對象要加入到TreeSet當中，那麼這個類必須要實現Comparable接口。

import java.util.Iterator;
import java.util.*;

public class TreeSetTest {
    public static void main(String[] args) {
        Set ts = new TreeSet();
        ts.add(new Teacher("zhangsan", 1));
        ts.add(new Teacher("lisi", 2));
        ts.add(new Teacher("wangmazi", 3));
        ts.add(new Teacher("mazi", 3));
        Iterator it = ts.iterator();
        while (it.hasNext()) {
            System.out.println(it.next());
        }
    }
}


class Teacher implements Comparable {
    int num;
    String name;

    Teacher(String name, int num) {
        this.num = num;
        this.name = name;
    }

    public String toString() {
        return "學好" + num + "姓名" + name;
    }

    public int compareTo(Object o) {
        Teacher ss = (Teacher) o;
        int result = num > ss.num ? 1 : (num == ss.num ? 0 : -1);
        if (result == 0) {
            result = name.compareTo(ss.name);
        }
        return result;
    }
}

注意：如果將int result = num > ss.num ? 1 : (num == ss.num ? 0 : -1);寫成了int result = ss.num > num ? 1 : (ss.num == num ? 0 : -1);那麼得到的結果就是倒序排列的，並不是升序的。

在使用Arrays對數組中的元素進行排序的時候，可以傳遞一個比較器。

在使用Collections對集合中的元素進行排序的時候，可以傳遞一個比較器。

那麼在使用TreeSet對加入到其中的元素進行排序的時候可以傳入一個比較器嗎？

public TreeSet(Comparator<? super E> comparator) {
this(new TreeMap<E,Object>(comparator));
}

通過查看它的構造方法就知道可以傳入一個比較器。

構造一個新的空TreeSet，它根據指定比較器進行排序。插入到該 set 的所有元素都必須能夠由指定比較器進行相互比較：對於 set 中的任意兩個元素 e1 和e2，執行 comparator.compare(e1, e2) 都不得拋出 ClassCastException。如果用戶試圖將違反此約束的元素添加到 set 中，則 add 調用將拋出 ClassCastException。

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

public class TreeSetTest {
public static void main(String[] args) {
TreeSet ts = new TreeSet(new Teacher.TeacherCompare());
ts.add(new Teacher("zhangsan", 2));
ts.add(new Teacher("lisi", 1));
ts.add(new Teacher("wangmazi", 3));
ts.add(new Teacher("mazi", 3));
Iterator it = ts.iterator();
while (it.hasNext()) {
System.out.println(it.next());
}
}
}

class Teacher {
int num;
String name;

Teacher(String name, int num) {
this.num = num;
this.name = name;
}

public String toString() {
return "學號：" + num + " 姓名：" + name;
}

static class TeacherCompare implements Comparator {// 老師自帶的一個比較器

public int compare(Object o1, Object o2) {
Teacher s1 = (Teacher) o1;// 轉型
Teacher s2 = (Teacher) o2;// 轉型
int result = s1.num > s2.num ? 1 : (s1.num == s2.num ? 0 : -1);
if (result == 0) {
result = s1.name.compareTo(s2.name);
}
return result;
}

}

} 復制代碼