diff --git a/blogs/集合/HashSet 和 TreeSet.md b/blogs/集合/HashSet 和 TreeSet.md new file mode 100644 index 0000000..a35947f --- /dev/null +++ b/blogs/集合/HashSet 和 TreeSet.md @@ -0,0 +1,50 @@ +--- +HashSet 和 TreeSet 源码解析 +--- + +#### HashSet + +HashSet 就是维护一个不包含重复元素的集合。 + +它的实现很简单,就是通过 HashMap 来做的。 + +```java +public class HashSet + extends AbstractSet + implements Set, Cloneable, java.io.Serializable +{ + private transient HashMap map; + private static final Object PRESENT = new Object(); + + public HashSet() { + map = new HashMap<>(); + } + + public boolean add(E e) { + return map.put(e, PRESENT)==null; + } +} +``` + +#### TreeSet + +TreeSet 大致的结构和 HashSet 相似,底层组合的是 TreeMap,所以继承了 TreeMap key 能够排序的功能,迭代的时候,也可以按照 key 的排序顺序进行迭代。 + +```java +public class TreeSet extends AbstractSet + implements NavigableSet, Cloneable, java.io.Serializable +{ + private transient NavigableMap m; + private static final Object PRESENT = new Object(); + + public TreeSet() { + this(new TreeMap<>()); + } + + public boolean add(E e) { + return m.put(e, PRESENT)==null; + } +} +``` + +但是和 HashSet 不同的是,TreeSet 可以实现从头或从尾进行遍历,这时是 TreeSet 定义接口,让 TreeMap 去实现的。 \ No newline at end of file