TreeMap的deleteEntry方法为何需判断p.color==BLACK后调用fixAfterDeletion？

首先得明确TreeMap底层是红黑树实现，红黑树的核心约束是所有从根到叶子的路径上，黑色节点的数量必须一致，删除操作如果破坏了这个约束，就需要调用fixAfterDeletion做修复。

你提到的replacement != null场景，对应被删除节点p只有一个非空子节点的情况（如果p有两个子节点，TreeMap会先用后继节点替换p，再删除后继节点，此时后继节点才会走到单节点/叶子的删除逻辑）。

先理清楚这个场景下的两种颜色组合：

• 若p是红色：根据红黑树规则，红色节点的子节点必须是黑色，所以replacement必然是黑色。把replacement移到p的位置后，路径的黑色节点数量完全没变——原来p是红色（不计入黑色高度），replacement本身就在路径里是黑色，现在只是往上挪了一层，总黑色节点数和之前一致，完全不破坏红黑树性质，所以不需要调用修复方法。
• 若p是黑色：这时候replacement只能是红色（如果replacement是黑色，那p所在路径的黑色节点数会比其他路径多1，不符合红黑树的前置合法状态）。把红色的replacement移到黑色p的位置后，这条路径的黑色节点数直接少了1（原来的p是黑色，现在换成红色的replacement不计入黑色高度），这就打破了红黑树的核心约束，所以必须调用fixAfterDeletion(replacement)——这个方法在这里的核心作用就是把replacement设为黑色，补回缺失的黑色节点，恢复路径的黑色高度平衡。

所以代码里先判断p.color == BLACK才调用修复方法，本质是只在删除操作真的破坏了红黑树性质时才触发修复，避免做无用功。如果p是红色，替换后性质没被破坏，自然不需要多此一举。

内容的提问来源于stack exchange，提问作者denis Krivorutchko