1110. 删点成林

删除二叉树中的节点，生成森林 —— 题解详解

题目描述

给定一棵二叉树的根节点 root，树上每个节点的值都是互不相同的。同时，给定一个整数列表 to_delete，表示需要从树中删除的节点值。

当删除一个节点时，它的子节点（如果存在）将成为新的树的根节点，形成一片森林（多棵互不相交的树的集合）。请你返回删除节点后形成的所有树的根节点列表。

输入

• root：二叉树根节点。
• to_delete：需要删除的节点值列表。

输出

• 返回删除指定节点后形成的森林中所有树的根节点列表。

---

题目分析

这道题的核心在于“删除节点”后，子节点如何成为新的树的根节点，形成森林。每当我们删除某个节点时：

• 该节点的子节点若存在，就成为新树的根节点。
• 原树结构断开，形成若干个不相交的子树。

与普通删除操作不同，删除节点后需要把子节点分离出来作为新的独立树。

---

解题思路

我们用递归深度优先搜索遍历整棵树，利用以下关键点：

1. 判断当前节点是否需要删除。
   预先将 to_delete 转成哈希集合