小編給大家分享一下Java中TreeSet的實現(xiàn)方法,希望大家閱讀完這篇文章后大所收獲,下面讓我們一起去探討吧!
HashSet是基于HashMap實現(xiàn)的,那TreeSet會是怎么實現(xiàn)的呢?沒錯!和大家想的一樣,它是基于TreeMap實現(xiàn)的。所以,TreeSet的源碼也很簡單,主要還是理解TreeMap。
TreeSet的繼承關(guān)系
按照慣例,先來看TreeSet類的繼承關(guān)系:
public class TreeSet<E> extends AbstractSet<E> implements NavigableSet<E>, Cloneable, java.io.Serializable
毫不意外的繼承了抽象類AbstracSet,方便擴展;
實現(xiàn)了一個NavigableSet接口,和NavigableMap接口類似,提供了各種導(dǎo)航方法;
實現(xiàn)了Cloneable接口,可以克隆;
實現(xiàn)了Serializable接口,可以序列化;
這里主要看NavigableSet接口類:
public interface NavigableSet<E> extends SortedSet<E>
熟悉的味道,繼承SortedSet接口。SortedSet則提供了一個返回比較器的方法:
Comparator<? super E> comparator();
和SortedMap一樣,支持自然排序和自定義排序。自然排序要求添加到Set中的元素實現(xiàn)Comparable接口,自定義排序要求實現(xiàn)一個Comparator比較器。
源碼分析
關(guān)鍵點
關(guān)鍵點自然是TreeSet如何保證元素不重復(fù)以及元素有序的,前面說了它是基于TreeMap實現(xiàn)的,那我們來看看吧。
private transient NavigableMap<E,Object> m; // 保證有序 // Dummy value to associate with an Object in the backing Map private static final Object PRESENT = new Object(); // 固定Value
縱觀TreeSet源碼,發(fā)現(xiàn)只有這兩個屬性(還有個uid,這里就不算了)。很明顯,m是用來保存元素的,但m聲明的是NavigableMap而不是TreeMap。可以猜測,TreeMap應(yīng)該是在構(gòu)造方法里實例化的,這里使用NavigableMap可以讓TreeSet更加靈活。PRESENT和HashSet中的PRESENT作用一樣,作為固定Value值進行占位的。
再看add和remove方法:
public boolean add(E e) { return m.put(e, PRESENT)==null; } public boolean remove(Object o) { return m.remove(o)==PRESENT; }
和HashSet的實現(xiàn)一樣,也是利用了Map保存的Key-Value鍵值對的Key不會重復(fù)的特點。
構(gòu)造函數(shù)
果然,TreeSet中的TreeMap是在構(gòu)造函數(shù)中初始化的。
public TreeSet() { this(new TreeMap<>()); // 默認自然排序的TreeMap } public TreeSet(Comparator<? super E> comparator) { this(new TreeMap<>(comparator)); // 自定義比較器的TreeMap } public TreeSet(Collection<? extends E> c) { this(); // 還是用的默認 addAll(c); // 將元素一個一個添加到TreeMap中 } public TreeSet(SortedSet<E> s) { this(s.comparator()); // 使用傳入的SortedSet的比較器 addAll(s); // 一個一個添加元素 }
默認實例化了一個自然排序的TreeMap,當(dāng)然,我們可以自定義比較器。
這里跟蹤下addAll方法:
public boolean addAll(Collection<? extends E> c) { // Use linear-time version if applicable if (m.size()==0 && c.size() > 0 && c instanceof SortedSet && m instanceof TreeMap) { SortedSet<? extends E> set = (SortedSet<? extends E>) c; TreeMap<E,Object> map = (TreeMap<E, Object>) m; // 強轉(zhuǎn)成TreeMap Comparator<?> cc = set.comparator(); Comparator<? super E> mc = map.comparator(); if (cc==mc || (cc != null && cc.equals(mc))) { // 要保證set和map的比較器一樣 map.addAllForTreeSet(set, PRESENT); // TreeMap專門為TreeSet準備的方法 return true; } } return super.addAll(c); }
調(diào)用了TreeMap的addAllForTreeSet
方法:
void addAllForTreeSet(SortedSet<? extends K> set, V defaultVal) { try { buildFromSorted(set.size(), set.iterator(), null, defaultVal); } catch (java.io.IOException | ClassNotFoundException cannotHappen) { } }
看到buildFromSorted,應(yīng)該很熟悉,在TreeMap的文章中分析過。該方法將傳入的集合元素構(gòu)造成了一棵最底層的結(jié)點為紅色,而其他結(jié)點都是黑色的紅黑樹。
導(dǎo)航方法
既然實現(xiàn)了NavigableSet,那各種導(dǎo)航方法自然少不了。它們的實現(xiàn)也很簡單,直接調(diào)用m對應(yīng)的導(dǎo)航方法即可。例如:
public E first() { return m.firstKey(); // 返回第一個元素 } public E lower(E e) { return m.lowerKey(e); // 返回小于e的第一個元素 } public NavigableSet<E> headSet(E toElement, boolean inclusive) { return new TreeSet<>(m.headMap(toElement, inclusive)); // 取前幾個元素構(gòu)成子集 } public E pollFirst() { // 彈出第一個元素 Map.Entry<E,?> e = m.pollFirstEntry(); return (e == null) ? null : e.getKey(); } public NavigableSet<E> descendingSet() { // 倒排Set return new TreeSet<>(m.descendingMap()); } ......
這里需要注意的是返回子集合的方法,例如:headSet。返回的子集合是可以添加和刪除元素的,但是有邊界限制,舉個栗子。
// 前面構(gòu)造了一個存儲Int的Set // 3、5、7、9 SortedSet<Integer> subSet = intSet.headSet(8); // 大值7,超過7越界 for (Integer sub : subSet) { System.out.println(sub); } subSet.add(2); // subSet.add(8); // 越界了 subSet.remove(3); for (Integer sub : subSet) { System.out.println(sub); }
TreeSet也是支持逆序輸出的,因為有descendingIterator
的實現(xiàn):
public Iterator<E> descendingIterator() { return m.descendingKeySet().iterator(); }
總結(jié)
TreeSet是基于TreeMap實現(xiàn)的,支持自然排序和自定義排序,可以進行逆序輸出;
TreeSet不允許null值;
TreeSet不是線程安全的,多線程環(huán)境下可以使用SortedSet s = Collections.synchronizedSortedSet(new TreeSet(...));
看完了這篇文章,相信你對Java中TreeSet的實現(xiàn)方法有了一定的了解,想了解更多相關(guān)知識,歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道,感謝各位的閱讀!
本文標題:Java中TreeSet的實現(xiàn)方法-創(chuàng)新互聯(lián)
文章路徑:http://vcdvsql.cn/article28/cccijp.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供網(wǎng)站建設(shè)、面包屑導(dǎo)航、網(wǎng)站策劃、微信公眾號、網(wǎng)站導(dǎo)航、網(wǎng)站制作
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請盡快告知,我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場,如需處理請聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時需注明來源: 創(chuàng)新互聯(lián)
猜你還喜歡下面的內(nèi)容