[数据结构与算法]RED-BLACK(红黑)树的实现TreeMap源码阅读

[数据结构与算法]RED-BLACK(红黑)树的实现TreeMap源码阅读
声明: 原创作品转载时请注明文章来自 SAP师太 技术博客( 博/客/园www.cnblogs.com):www.cnblogs.com/jiangzhengjun并以超链接形式标明文章原始出处否则将追究法律责任!原文链接:http://www.cnblogs.com/jiangzhengjun/p/4289768.html

由于平衡二叉树与红黑树都是二叉排序树,又红黑树是对平衡二叉树的一种改进实现,所以它的很多思想算法都来源于排序二叉或平衡二叉树,比如排序二叉树中的添加、删除、查找及查找直接后继节点等,平衡二叉树中的左旋与右旋等都是一样的,所以当看到这些方法时,要多参考以下两节:二叉排序(搜索)树实现平衡二叉树实现

 

SortedMap 接口的基于红黑树的实现。此类保证了Map按照升序顺序排列关键字,根据使用的构造方法不同,可能会按照键的类的自然顺序进行排序 (Comparable),或者按照创建时所提供的比较(Comparator)进行排序。此实现不是同步的,多个线程同时访问一个同一个map时,一般通过对封装该map的对象进行同步操作来完成,或者使用  Map m = Collections.synchronizedMap(new TreeMap(...)) 方法来包装该map

RED-BLACK树的性质:
1、根元素永远是显色。
2、除根节点外,所有新插入的节点都是红色的。


3、Red规则:红色节点的子节点都是黑色的,不能有两个红色节点相邻。
4、Path规则:从根元素到某个子节点或是只有一个子节点的元素的所有路径中,黑色元素的个数必须相同。

 

  1 package tree.redblack;

  2 

  3 import java.util.AbstractCollection;

  4 import java.util.AbstractMap;

  5 import java.util.AbstractSet;

  6 import java.util.Collection;

  7 import java.util.Comparator;

  8 import java.util.ConcurrentModificationException;

  9 import java.util.Iterator;

 10 import java.util.Map;

 11 import java.util.NoSuchElementException;

 12 import java.util.Set;

 13 import java.util.SortedMap;

 14 

 15 public class TreeMap extends AbstractMap implements SortedMap, Cloneable,

 16         java.io.Serializable {

 17 

 18     //比较器,用于TreeMap的key的排序,如果为null,则使用key的自然排序法

 19     private Comparator comparator = null;

 20 

 21     private transient Entry root = null;//根节点

 22 

 23     private transient int size = 0;//树中的节点数

 24 

 25     /*

 26      * TreeMap结构被修改的次数,注:结构上修改是指添加或删除一个或多个映射关系的操作, 仅改变与

 27      * 现有键关联的值不是结构上的修改。

 28      * 

 29      * 由所有此类的 collection 视图方法(如keySet、values、entrySet三个方法) 所返回的迭代

 30      * 器都是快速失效的:在迭代器创建之后,如果对map修改了结构,除非通过迭代器自身的 remove 或

 31      *  add 方法,其他map自身任何时间任何方式的修改,迭代器都将抛出 ConcurrentModification

 32      *  Exception。因此,面对并发的修改,迭代器很快就完全失效,而不是冒着在将来在不确定的时间点

 33      *  上发生不确定问题的风险。 

 34      */

 35     private transient int modCount = 0;

 36 

 37     //增加节点后modCount与size都需递增

 38     private void incrementSize() {

 39         modCount++;

 40         size++;

 41     }

 42 

 43     private void decrementSize() {

 44         modCount++;

 45         size--;

 46     }

 47 

 48     /**

 49      * 默认构造器,构造一个空的map,根据key的自然比较排序。所有插入到map中的节点的key必须实现

 50      * 过自然比较器Comparable接口。

 51      */

 52     public TreeMap() {

 53     }

 54 

 55     /**

 56      * 构造一个新的空映射,该映射根据给定的比较器进行排序

 57      */

 58     public TreeMap(Comparator c) {

 59         this.comparator = c;

 60     }

 61 

 62     /**

 63      * 构造一个新map,所含的元素与给定的map相同,这个新map按照键的自然顺序进行关键字排序

 64      */

 65     public TreeMap(Map m) {

 66         putAll(m);

 67     }

 68 

 69     /**

 70      * 构造一个新的映射,所含的元素与给定的map相同,该映射按照SortedMap相同的排序方式进行排序

 71      */

 72     public TreeMap(SortedMap m) {

 73         comparator = m.comparator();

 74         //...

 75     }

 76 

 77     public int size() {

 78         return size;

 79     }

 80 

 81     public boolean containsKey(Object key) {

 82         return getEntry(key) != null;

 83     }

 84 

 85     public boolean containsValue(Object value) {

 86         return (root == null ? false : (value == null ? valueSearchNull(root)

 87                 : valueSearchNonNull(root, value)));

 88     }

 89 

 90     //树中是否有value域值为null的节点

 91     private boolean valueSearchNull(Entry n) {

 92         if (n.value == null)

 93             return true;

 94 

 95         // 递归在左右子树中查找

 96         return (n.left != null && valueSearchNull(n.left))

 97                 || (n.right != null && valueSearchNull(n.right));

 98     }

 99 

100     /*

101      * 查找指定节点的value值域的节点,因为树是按节的key关键字来排序的,而不是按value排序的,

102      * 所以在找value时在最坏的情况下遍历整个树

103      * 

104      * 以二叉树的先序遍历查找

105      */

106     private boolean valueSearchNonNull(Entry n, Object value) {

107         // 先比较根

108         if (value.equals(n.value))

109             return true;

110 

111         // 再比较左,最后再比较右

112         return (n.left != null && valueSearchNonNull(n.left, value))

113                 || (n.right != null && valueSearchNonNull(n.right, value));

114     }

115 

116     //获取指定key的值

117     public Object get(Object key) {

118         Entry p = getEntry(key);

119         return (p == null ? null : p.value);

120     }

121 

122     //返回树中第一个(最小的)键,最左边的节点

123     public Object firstKey() {

124         return key(firstEntry());

125     }

126 

127     /**

128      * Returns the last (highest) key currently in this sorted map.

129      *

130      * @return the last (highest) key currently in this sorted map.

131      * @throws    NoSuchElementException Map is empty.

132      */

133     public Object lastKey() {

134         return key(lastEntry());

135     }

136 

137     /*

138      * 根据给定的key查找节点,因为 RED-BLACK树也是一种二叉排序树,所以采用在二叉排序树

139      * 中查找节点的算法来查找

140      */

141     private Entry getEntry(Object key) {

142         Entry p = root;

143         while (p != null) {

144             int cmp = compare(key, p.key);

145             if (cmp == 0)

146                 return p;//找到则返回

147             else if (cmp < 0)

148                 p = p.left;//如果关键字小于当前节点,则在当前节点的左子树中找

149             else

150                 p = p.right;//如果关键字大于当前节点,则在当前节点的右子树中找

151         }

152         return null;

153     }

154 

155     private static Object key(Entry e) {

156         if (e == null)

157             throw new NoSuchElementException();

158         return e.key;

159     }

160 

161     /**

162      * 如果key已存在,将替换原值,并返回原来key所对应的值;如果不存在,则返回 null(注,如果

163      * 新增key已存在,且为null,返回时也会为null,value为null并不代表key不存在)

164      */

165     public Object put(Object key, Object value) {

166         //插入根节点时不需要调整颜色

167         Entry t = root;

168 

169         if (t == null) {

170             incrementSize();

171             root = new Entry(key, value, null);

172             return null;

173         }

174 

175         while (true) {

176             int cmp = compare(key, t.key);

177             if (cmp == 0) {

178                 return t.setValue(value);

179             } else if (cmp < 0) {

180                 if (t.left != null) {

181                     t = t.left;

182                 } else {

183                     incrementSize();

184                     t.left = new Entry(key, value, t);

185                     //需要调整颜色

186                     fixAfterInsertion(t.left);

187                     return null;

188                 }

189             } else { // cmp > 0

190                 if (t.right != null) {

191                     t = t.right;

192                 } else {

193                     incrementSize();

194                     t.right = new Entry(key, value, t);

195                     //需要调整颜色

196                     fixAfterInsertion(t.right);

197                     return null;

198                 }

199             }

200         }

201     }

202 

203     //从map中删除指定的key节点,并返回value,如果不存在或value本身就是null时,返回null

204     public Object remove(Object key) {

205         Entry p = getEntry(key);

206         if (p == null)

207             return null;

208 

209         Object oldValue = p.value;

210         deleteEntry(p);

211         return oldValue;

212     }

213 

214     public void clear() {

215         modCount++;

216         size = 0;

217         root = null;

218     }

219 

220     public Object clone() {

221         TreeMap clone = null;

222         try {

223             //调整父类的克隆方法

224             clone = (TreeMap) super.clone();

225         } catch (CloneNotSupportedException e) {

226             throw new InternalError();

227         }

228 

229         // Put clone into "virgin" state (except for comparator)

230         clone.root = null;

231         clone.size = 0;

232         clone.modCount = 0;

233         clone.entrySet = null;

234         //...

235         return clone;

236     }

237 

238     // Views

239 

240     /**

241      * key-value(键-值对)视图,即该set视图里是一个个的Entry实体节点

242      */

243     private transient volatile Set entrySet = null;

244 

245     //以下两个字段是从父类AbstractMap拷贝过来的,因为在该类与父类不在一个包中,所以不可见

246     transient volatile Set keySet = null;//key视图,set视图里是一个个Entry实体节点的key

247     //value视图,collection视图里是一个个Entry实体节点的value

248     transient volatile Collection values = null;

249 

250     public Set keySet() {

251         if (keySet == null) {

252             //keySet视图实现

253             keySet = new AbstractSet() {

254                 public Iterator iterator() {

255                     //只能对key进行迭代

256                     return new KeyIterator();

257                 }

258 

259                 public int size() {

260                     return TreeMap.this.size();

261                 }

262                 //...

263             };

264         }

265         return keySet;

266     }

267 

268     //value视图中的value排列的顺序保持不变,实质上还是在entrySet视图上迭代的

269     public Collection values() {

270         if (values == null) {

271             //value视图实现

272             values = new AbstractCollection() {

273                 public Iterator iterator() {

274                     //只能迭代value

275                     return new ValueIterator();

276                 }

277 

278                 public int size() {

279                     return TreeMap.this.size();

280                 }

281                 //...

282             };

283         }

284         return values;

285     }

286 

287     //key-value 视图,即Entry节点视图

288     public Set entrySet() {

289         if (entrySet == null) {

290             entrySet = new AbstractSet() {

291                 public Iterator iterator() {

292                     //对Entry节点迭代

293                     return new EntryIterator();

294                 }

295 

296                 //...

297                 public int size() {

298                     return TreeMap.this.size();

299                 }

300 

301             };

302         }

303         return entrySet;

304     }

305 

306     //Entry迭代器

307     private class EntryIterator implements Iterator {

308         private int expectedModCount = TreeMap.this.modCount;

309         private Entry lastReturned = null;//指向最后一次next操作所返回的节点

310         Entry next;//指向当前next()操作要返回的节点

311 

312         EntryIterator() {

313             next = firstEntry();//开始时next指向最左边的节点,即中序遍历序列中的第一个节点

314         }

315 

316         // Used by SubMapEntryIterator

317         EntryIterator(Entry first) {

318             next = first;

319         }

320 

321         public boolean hasNext() {

322             //如果next没有指向null,则表示当前next指向的节点还有

323             return next != null;

324         }

325 

326         //KeyIterator与ValueIterator两迭代器的next()方法实现都是调整此方法来实现的

327         final Entry nextEntry() {

328             if (next == null)

329                 throw new NoSuchElementException();

330             if (modCount != expectedModCount)

331                 throw new ConcurrentModificationException();

332             lastReturned = next;//先取

333             next = successor(next);//next再下移

334             return lastReturned;

335         }

336 

337         public Object next() {

338             //对Entry进行迭代

339             return nextEntry();

340         }

341 

342         public void remove() {

343             if (lastReturned == null)

344                 throw new IllegalStateException();

345             if (modCount != expectedModCount)

346                 throw new ConcurrentModificationException();

347             //如果删除的节点左右子树都存在,则删除后next指针需后退到lastReturned的位置,

348             //具体为什么,请参考二叉搜索树的实现 BinSearchTree.java 相应方法

349             if (lastReturned.left != null && lastReturned.right != null)

350                 next = lastReturned;

351             deleteEntry(lastReturned);

352             expectedModCount++;

353             lastReturned = null;

354         }

355     }

356 

357     //KeyIterator是在EntryIterator的基础上只对key进行迭代罢了

358     private class KeyIterator extends EntryIterator {

359         public Object next() {

360             return nextEntry().key;

361         }

362     }

363 

364     //ValueIterator是在EntryIterator的基础上只对value进行迭代罢了

365     private class ValueIterator extends EntryIterator {

366         public Object next() {

367             return nextEntry().value;

368         }

369     }

370 

371     //在这里决定是使用 Comparator 还 Comparable 比较器

372     private int compare(Object k1, Object k2) {

373         return (comparator == null ? ((Comparable) k1).compareTo(k2) : comparator

374                 .compare(k1, k2));

375     }

376 

377     private static boolean valEquals(Object o1, Object o2) {

378         return (o1 == null ? o2 == null : o1.equals(o2));

379     }

380 

381     private static final boolean RED = false;//false代表红

382     private static final boolean BLACK = true;//true代表黑

383 

384     /**

385      * 树中的节点结构体实现,实现了Map.Entry接口,它是双向的,即可以从子找到父,也可从

386      * 父到子节点

387      */

388 

389     static class Entry implements Map.Entry {

390         Object key;//数据域key

391         Object value;//数据域value

392         Entry left = null;//左指针

393         Entry right = null;//右指针

394         Entry parent;//父指针

395         boolean color = BLACK;//节点着色,相当于平衡二叉排序树中的平衡因子

396 

397         Entry(Object key, Object value, Entry parent) {

398             this.key = key;

399             this.value = value;

400             this.parent = parent;

401         }

402 

403         public Object getKey() {

404             return key;

405         }

406 

407         public Object getValue() {

408             return value;

409         }

410 

411         public Object setValue(Object value) {

412             Object oldValue = this.value;

413             this.value = value;

414             return oldValue;

415         }

416 

417         //...

418         public String toString() {

419             return key + "=" + value;

420         }

421     }

422 

423     //取树中最左边的节点,即key最小的节点

424     private Entry firstEntry() {

425         Entry p = root;

426         if (p != null)

427             while (p.left != null)

428                 p = p.left;

429         return p;

430     }

431 

432     //取树中最右边的节点,即key最大的节点

433     private Entry lastEntry() {

434         Entry p = root;

435         if (p != null)

436             while (p.right != null)

437                 p = p.right;

438         return p;

439     }

440 

441     //查找某节点的中序遍历的直接后继节点,具体请参见BinSearchTree.java 相应方法

442     private Entry successor(Entry t) {

443         if (t == null)

444             return null;

445         else if (t.right != null) {

446             Entry p = t.right;

447             while (p.left != null)

448                 p = p.left;

449             return p;

450         } else {

451             Entry p = t.parent;

452             Entry ch = t;

453             while (p != null && ch == p.right) {

454                 ch = p;

455                 p = p.parent;

456             }

457             return p;

458         }

459     }

460 

461     //平衡操作

462     private static boolean colorOf(Entry p) {

463         //注,如果为null,则返回为黑色,即如果求某个节点的子节点颜色,但子节点不存在时也会返回黑色

464         return (p == null ? BLACK : p.color);

465     }

466 

467     private static Entry parentOf(Entry p) {

468         return (p == null ? null : p.parent);

469     }

470 

471     private static void setColor(Entry p, boolean c) {

472         if (p != null)

473             p.color = c;

474     }

475 

476     private static Entry leftOf(Entry p) {

477         return (p == null) ? null : p.left;

478     }

479 

480     private static Entry rightOf(Entry p) {

481         return (p == null) ? null : p.right;

482     }

483 

484     //左旋,具体请参见AVLTree.java相应方法

485     private void rotateLeft(Entry p) {

486         Entry r = p.right;

487         p.right = r.left;

488         if (r.left != null)

489             r.left.parent = p;

490         r.parent = p.parent;

491         if (p.parent == null)

492             root = r;

493         else if (p.parent.left == p)

494             p.parent.left = r;

495         else

496             p.parent.right = r;

497         r.left = p;

498         p.parent = r;

499     }

500 

501     //右旋,具体请参见AVLTree.java相应方法

502     private void rotateRight(Entry p) {

503         Entry l = p.left;

504         p.left = l.right;

505         if (l.right != null)

506             l.right.parent = p;

507         l.parent = p.parent;

508         if (p.parent == null)

509             root = l;

510         else if (p.parent.right == p)

511             p.parent.right = l;

512         else

513             p.parent.left = l;

514         l.right = p;

515         p.parent = l;

516     }

517 

518     /** From CLR **/

519     private void fixAfterInsertion(Entry x) {

520         //插入的节点为红色

521         x.color = RED;

522 

523         /*

524          * 添加节点时,因为在上面将新增节点设置成了红色,所以添加后是不会打破 Path规则 的,但是

525          * 有可能会打破 Red规则,又打破 Red规则 的前提是父节点是红色的才可能,所以这里的循环条

526          * 件就是父节点必须是红色的才需调整

527          * 

528          * 1、如果x为根,则不需要重新染色与旋转,因为最后会将根再次染成BLACK

529          * 2、如果x的父节点color为黑色,也没必要进行重排了,因为red规则没有被破坏

530          * 如果上面两个条件不满足,则一直循环,这里x != null 条件也是不可少的,传递进来的值

531          * 与循环终止时都能可能为null

532          */

533         while (x != null && x != root && x.parent.color == RED) {

534 

535             //一、x的父节点处于左子节点时

536             if (parentOf(x) == leftOf(parentOf(parentOf(x)))) {

537 

538                 //y指向x的右叔,注,有可能为NULL

539                 Entry y = rightOf(parentOf(parentOf(x)));

540                 //如果y为红色

541                 if (colorOf(y) == RED) {//如果条件成立,则y不能可为空

542                     /*

543                      * 情况1:colorOf(y) == RED

544                      * 

545                      * 此情况下y不可能为null,因为 colorOf(null) == BLACK,所以此情况

546                      * 下右叔一定存在,既然右叔是红色,所以爷爷节点50只能是黑色了,否则打破

547                      * 了Red规则了。30也肯定是红色了,因为进行while循环前提条件就是父节点

548                      * 一定是红色的

549                      *                         着色

550                      *           .              →              .

551                      *           .                             .

552                      *           50B                      x → 50R

553                      *           /\                            /\

554                      *        30R  90R ← y(右叔)             30B  90B ← y

555                      *           \                            \

556                      *      x → 40R                          40R

557                      *      

558                      * 此情况下不需要旋转,完成着色后继续循环

559                      */

560 

561                     //将x节点的父节点的着黑

562                     setColor(parentOf(x), BLACK);

563                     //将x的右叔的着黑

564                     setColor(y, BLACK);

565                     //将x的爷爷着红

566                     setColor(parentOf(parentOf(x)), RED);

567 

568                     /*

569                      * 完成操作后,我们发现x向上移动了两层,因此循环的最大次数是树的高度的一

570                      * 半,这也就是TreeMap中的put方法的最大执行时间为 O(logn) 原因所在了

571                      */

572                     x = parentOf(parentOf(x));//再将x指向爷爷

573                 }//如果y为黑色

574                 else {

575 

576                     if (x == rightOf(parentOf(x))) {

577                         /*

578                         * 情况2:colorOf(y) == BLACK,且x为父的右子节点

579                         * 

580                         * 此情况下,y一定是null,为什么?因为插入的节点40的父节点30为红色

581                         * (又为什么?因为进行while循环的前提就是父为红色啊),所以30的父

582                         * 节点只能为黑色了(又为什么?因为如果是红色的话,早就打破了Red规则

583                         * 了),现进入到此分支时前提是y必须是黑色的,如果y真的存在,则Path

584                         * 规则又将被打破,所以此情况下的y一定是null,这样才满足Path规则

585                         * 

586                         * ====实质上此种情况就是将问题2转换成转换成问题3来操作====

587                         * 

588                         *                   绕30左旋

589                         *        .            →             .

590                         *        .                          .

591                         *       50B                        50B

592                         *        /                          / 

593                         *      30R  (Y为NULL)             40R  

594                         *        \                         /

595                         *   x → 40R                   x → 30R

596                         */

597                         x = parentOf(x);

598                         rotateLeft(x);

599                     }

600 

601                     /*

602                      * 情况3:colorOf(y) == BLACK,且x为父的左子节点

603                      * 

604                      * 如果x为父的右子节点,经过上面情况2的处理,已经将其转换成情况3了,此种

605                      * 情况下的y也一定为null,与第2种情况道理是一样的

606                      * 

607                      *                    着色               绕50右旋

608                      *        .            →         .       →       .

609                      *        .                      .               .

610                      *       50B                    50R            40B

611                      *        /                      /               /\

612                      *      40R  (Y为NULL)          40B        x → 30R 50R

613                      *       /                      /

614                      * x → 30R                 x → 30R

615                      * 

616                      * 处理后,下次循环自动会终止,因为此情况处理后x的父节点总是黑色的,所以

617                      * 循环一定会终止于下次

618                      */

619                     setColor(parentOf(x), BLACK);//x的父着黑

620                     setColor(parentOf(parentOf(x)), RED);//x的爷着红

621                     if (parentOf(parentOf(x)) != null)

622                         rotateRight(parentOf(parentOf(x)));//绕爷爷右旋

623                 }

624             }//二、x的父节点处于右子节点时,与第一种对称

625             else {

626                 //x的左叔

627                 Entry y = leftOf(parentOf(parentOf(x)));

628                 //如果左叔为红色

629                 if (colorOf(y) == RED) {

630                     /*

631                     * 情况1:colorOf(y) == RED,即左叔为红色

632                     * 

633                     * 此情况下y不可能为null,因为 colorOf(null) == BLACK,所以此情况

634                     * 下左叔一定存在,既然左叔是红色,所以爷爷节点50只能是黑色了,否则打破

635                     * 了Red规则了。90也肯定是红色了,因为进行while循环前提条件就是父节点

636                     * 一定是红色的

637                     *                         着色

638                     *           .              →              .

639                     *           .                             .

640                     *           50B                      x → 50R

641                     *           /\                            /\

642                     * y(左叔)→30R  90R                      30B 90B ← y

643                     *              \                             \

644                     *          x → 100R                          100R

645                     *      

646                     * 此情况下不需要旋转,完成着色后继续循环

647                     */

648                     setColor(parentOf(x), BLACK);

649                     setColor(y, BLACK);

650                     setColor(parentOf(parentOf(x)), RED);

651                     /*

652                      * 完成操作后,我们发现x向上移动了两层,因此循环的最大次数是树的高度的一

653                      * 半,这也就是TreeMap中的put方法的最大执行时间为 O(logn) 原因所在了

654                      */

655                     x = parentOf(parentOf(x));//再将x指向爷爷

656                 }//如果y为黑色

657                 else {

658                     if (x == leftOf(parentOf(x))) {

659                         /*

660                         * 情况2:colorOf(y) == BLACK(左叔为黑),且x为父的左子节点

661                         * 

662                         * 此情况下,y一定是null,为什么?因为插入的节点60的父节点90为红色

663                         * ,所以90的父节点只能为黑色了,现进入到此分支时前提是y必须是黑色的

664                         * ,如果y真的存在,则Path规则又将被打破,所以此情况下的y一定是null

665                         * ,这样才满足Path规则

666                         * 

667                         * ====实质上此种情况就是将问题2转换成转换成问题3来操作====

668                         * 

669                         *                   绕90右旋

670                         *          .            →             .

671                         *          .                          .

672                         *          50B                        50B

673                         *           \                          \ 

674                         *(Y为NULL)  90R                        60R  

675                         *           /                           \

676                         *      x → 60R                      x → 90R

677                         */

678                         x = parentOf(x);

679                         rotateRight(x);

680                     }

681 

682                     /*

683                      * 情况3:colorOf(y) == BLACK,且x为父的右子节点

684                      * 

685                      * 如果x为父的左子节点,经过上面情况2的处理,已经将其转换成情况3了,此种

686                      * 情况下的y也一定为null,与第2种情况道理是一样的

687                      * 

688                      *                    着色               绕50左旋

689                      *         .            →         .       →       .

690                      *         .                      .               .

691                      *        50B                    50R            90B

692                      *         \                      \               /\

693                      *(Y为NULL) 90R                   90B       x → 50R 95R

694                      *           \                     \

695                      *       x → 95R               x → 95R

696                      * 

697                      * 处理后,下次循环自动会终止,因为此情况处理后x的父节点总是黑色的,所以

698                      * 循环一定会终止于下次

699                      */

700                     setColor(parentOf(x), BLACK);//x的父着黑

701                     setColor(parentOf(parentOf(x)), RED);//x的爷着红

702                     if (parentOf(parentOf(x)) != null)

703                         rotateLeft(parentOf(parentOf(x)));//绕爷爷左旋

704                 }

705             }

706         }

707         root.color = BLACK;//不管怎样调整,最后根节点都要着黑

708     }

709 

710     //删除节点p,然后重新调整

711     private void deleteEntry(Entry p) {

712         decrementSize();

713 

714         /*

715         * 如果删除的节点p有左右子树时,将问题转换成删除叶子节点或只有一个子节点的节点问题,具体

716         * 过程:使用中序遍历直接后s继的数据域值(key与value)拷贝到p的相应数据域值,然后将p指

717         * 向直接后继s,这样待删除的节点就变成了s,问题已转换成删除叶节点或只有一个子节点的问题了

718         */

719         if (p.left != null && p.right != null) {

720             Entry s = successor(p);

721             p.key = s.key;

722             p.value = s.value;

723             p = s;

724         }

725 

726         //replacement用来替换即将被删除节点p

727         Entry replacement = (p.left != null ? p.left : p.right);

728 

729         //如果待删除的元素有子节点(如果有,则也有且仅有一个子节点)

730         if (replacement != null) {

731             // 重新设置替换节点的父

732             replacement.parent = p.parent;

733             //如果待删除节点的为根节点

734             if (p.parent == null)

735                 root = replacement;//则替换元素将成为根

736             else if (p == p.parent.left)

737                 //否则如果待删除节点为左节点,则使待删除节点的父的左指针指向替换节点

738                 p.parent.left = replacement;

739             else

740                 //否则如果待删除节点为右节点,则使待删除节点的父的右指针指向替换节点

741                 p.parent.right = replacement;

742 

743             // 解除p与其他节点的关系

744             p.left = p.right = p.parent = null;

745 

746             /*

747              * 因为被删除的节点只有两种节点: 叶子节点、只有一个子节点的节点

748              * 

749              * 如果删除的是只有一个子节点的节点时,根据路径规则,被删除节点一定是黑色的,而子节点

750              * 一定是红色的,这时删除操作过程就是删除了一个黑红节点而在删除位置上又放置了一个红色

751              * 节点,因此,删除一个有一个子节点的节点时,Red与Path规则都有可能打破

752              */

753 

754             if (p.color == BLACK)//???这里的条件好像是多余的,因为p一定是黑,难道另有玄机?

755                 //删除后从替换节点replacement位置向根方向调整

756                 fixAfterDeletion(replacement);

757         } else if (p.parent == null) { // 如果待删除节点为根,则置root为null

758             root = null;

759         } else { // 如果删除的是叶子节点

760 

761             /*

762              * 如果删除的是叶子节点,不管该叶子节点是红还是黑,则 Red规则 在删除后不可能被打破,

763              * 所以此情况下只有可能打破 Path规则 ,但打破的前提是被删除的节点是黑色的,因为只有

764              * 黑色节点才影响Path路径规则,所以调整的前提是 p.color == BLACK,即被删除的叶

765              * 节点是黑色的

766              */

767             if (p.color == BLACK)

768                 //删除前从即将被删除节点p位置向根方向调整

769                 fixAfterDeletion(p);

770 

771             //???此条件好像多余,因为如果不成立,则走上面那个分支了,难道另藏玄机

772             if (p.parent != null) {

773                 if (p == p.parent.left)

774                     p.parent.left = null;

775                 else if (p == p.parent.right)

776                     p.parent.right = null;

777                 p.parent = null;//解除p与其他节点的关系

778             }

779         }

780     }

781 

782     //删除节点后调整

783     private void fixAfterDeletion(Entry x) {

784 

785         //直循环到根节点或红色止

786         while (x != root && colorOf(x) == BLACK) {

787             //x为左子节点

788             if (x == leftOf(parentOf(x))) {

789                 //x的同胞

790                 Entry sib = rightOf(parentOf(x));//有可能为null

791 

792                 if (colorOf(sib) == RED) {//同胞不可能为null

793                     /*

794                      * 情况1:colorOf(sib) == RED,即右叔存在且为红

795                      *              通过执行上面deleteEntry后,x、p均指向后继65

796                      *        p → 60B             →           65B

797                      *            /  \                        /  \

798                      *          50B   70B                   50B   70B

799                      *          /\    /  \                  /\    /  \

800                      *       45B 55B 65B 80R             45B 55B 65B 80R ← sib

801                      *                   / \                     ↑   /  \

802                      *                 75B 87B                 x、p 75B 87B

803                      *  着色                         绕70左旋  

804                      *   →        65B               →           65B

805                      *            /  \                          /  \

806                      *          50B   70R                     50B   80B

807                      *          /\    /  \                    /\    / \

808                      *       45B 55B 65B 80B ← sib         45B 55B 70R 87B 

809                      *                ↑   / \                      /  \

810                      *              x、p 75B 87B            x、p → 65B 75B ← sib

811                      *  x现有一个新的同胞节点,该节点为黑,现将情况1转向其他3种情况

812                      */

813 

814                     setColor(sib, BLACK);//置同胞为黑

815                     setColor(parentOf(x), RED);//置父为红

816                     rotateLeft(parentOf(x));//左旋

817                     sib = rightOf(parentOf(x));//再处理同胞节点

818                 }

819 

820                 if (colorOf(leftOf(sib)) == BLACK && colorOf(rightOf(sib)) == BLACK) {

821                     /*

822                     * 情况2:

823                     * colorOf(leftOf(sib))==BLACK && colorOf(rightOf(sib))==BLACK

824                     * 即同胞不存在或同胞存在时为黑且没有子节点的情况

825                     * 

826                     * 现接着上面情况1结束后的那棵树处理

827                     * 着色                       再将x指向父(70R)继续处理

828                     *  →      65B                   →           ...

829                     *         /  \

830                     *       50B   80B

831                     *       /\     / \

832                     *     45B 55B 70R 87B

833                     *             / \

834                     *     x、p → 65B 75R ← sib

835                     * 

836                     * 处理完后,x为红,因此结束循环,然后设置x的颜色为黑,并从此返回

837                     * deleteEntry方法,此时p所指向的节点已被从其父节点解除链接

838                     */

839                     setColor(sib, RED);

840                     x = parentOf(x);

841                 } else {

842                     if (colorOf(rightOf(sib)) == BLACK) {

843                         /*

844                         * 情况3:经过情况2后,此时同胞节点左子节点必为红

845                         * 

846                         * 假设在开始while循环之前有如下树:

847                         *  →      65B                   

848                         *         /  \

849                         *       50B   80R

850                         *      / \    /  \

851                         *    45B 55B 70B  90B

852                         *            / \   / \

853                         *     x → 65B 75B 85R 95B

854                         *             ↗  / \

855                         *          sib  82B 87B

856                         *          

857                         * 情况2被应用,因此将75的颜色设置为红,并令 x = parentOf(x)后第

858                         * 一遍循环结束。第二遍循环开始时,会应用情况3,此时同胞节点为90,树

859                         * 如下图:

860                         * 第一次循环完                  着色

861                         *    →    65B                →          65B    

862                         *        /   \                          /  \

863                         *      50B    80R                    50B    80R

864                         *     / \     /  \                   / \    /  \

865                         *   45B 55B 70B   90B ← sib       45B 55B 70B   90R ← sib

866                         *         ↗/ \    / \                  ↗/ \     / \               

867                         *       x 65B 75R 85R 95B             x 65B 75R 85B 95B

868                         *                 / \                           / \

869                         *               82B 87B                       82B 87B

870                         *               

871                         * 绕90右旋后并修改sib指向           进行下一步循环,应用情况4

872                         *    →          65B                      → ...  

873                         *              /   \                        

874                         *            50B    80R                   

875                         *           / \     /  \                 

876                         *         45B 55B 70B   85B ← sib        

877                         *               ↗/ \    / \                

878                         *             x 65B 75R 82B 90R           

879                         *                           / \                         

880                         *                         87B 95B            

881                         *                         

882                         * 一旦情况3被应用,则同胞节点的右子节点将设置为 RED,因此情况4在情况

883                         * 3之后应用,为了不在下一次循环中应用情况4,直接将情况4的代码放在了

884                         * 情况3的后面

885                         */

886                         setColor(leftOf(sib), BLACK);

887                         setColor(sib, RED);

888                         rotateRight(sib);

889                         sib = rightOf(parentOf(x));

890                     }

891                     /*

892                      * 情况4:经过情况3后剩下的情况4,此时同胞节点的右子节点是红

893                      * 

894                      * 接着第3种情况处理:

895                      * 

896                      *   着色                    绕80左旋

897                      *    →       65B            →              65B

898                      *           /   \                          /  \

899                      *         50B    80B                     50B   85R ← sib

900                      *        / \     /  \                   / \    /  \

901                      *      45B 55B 70B   85R ← sib        45B 55B 80B 90B        

902                      *            ↗/ \    / \                     / \   \ 

903                      *          x 65B 75R 82B 90B            x → 70B 87B 95B

904                      *                        / \                 /

905                      *                      87B 95B         p → 65B

906                      *                      

907                      * 循环结束后返回到 deleteEntry 方法,删除65节点即完成删除操作过程

908                      */

909                     setColor(sib, colorOf(parentOf(x)));

910                     setColor(parentOf(x), BLACK);

911                     setColor(rightOf(sib), BLACK);

912                     rotateLeft(parentOf(x));

913 

914                     //情况3与4被应用后,x将被设置为根节点,同时,这也是循环的最后一次

915                     x = root;

916                 }

917             } else { // 与上面对称

918                 Entry sib = leftOf(parentOf(x));

919 

920                 if (colorOf(sib) == RED) {

921                     setColor(sib, BLACK);

922                     setColor(parentOf(x), RED);

923                     rotateRight(parentOf(x));

924                     sib = leftOf(parentOf(x));

925                 }

926 

927                 if (colorOf(rightOf(sib)) == BLACK && colorOf(leftOf(sib)) == BLACK) {

928                     setColor(sib, RED);

929                     x = parentOf(x);

930                 } else {

931                     if (colorOf(leftOf(sib)) == BLACK) {

932                         setColor(rightOf(sib), BLACK);

933                         setColor(sib, RED);

934                         rotateLeft(sib);

935                         sib = leftOf(parentOf(x));

936                     }

937                     setColor(sib, colorOf(parentOf(x)));

938                     setColor(parentOf(x), BLACK);

939                     setColor(leftOf(sib), BLACK);

940                     rotateRight(parentOf(x));

941                     x = root;

942                 }

943             }

944         }

945         //退出循环时x为根或为红色,直接将其值为黑色即可

946         setColor(x, BLACK);

947     }

948 }

你可能感兴趣的:(TreeMap)