删除一棵树

假设树的节点定义如下： ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right ``` 可以考虑使用深度优先搜索（DFS）的方式进行遍历和删除。 具体而言，我们可以从根节点开始进行深度优先搜索，并记录下访问的节点。当我们遍历到一个叶子节点时，就将其删除，并返回其父节点。如果父节点的左右子节点都被删除了，那么我们也将其删除，并返回其父节点。如此往复，直到整棵树都被删除完。 下面是具体的实现： ```python def del_leaf_nodes(root): if not root: return [] res = [] stack = [root] while stack: node = stack[-1] if not node.left and not node.right: res.append(node.val) stack.pop() if stack: parent = stack[-1] if parent.left == node: parent.left = None else: parent.right = None else: if node.right: stack.append(node.right) node.right = None if node.left: stack.append(node.left) node.left = None return res ``` 其中，`res` 用于存放删除的顺序，`stack` 用于存放待访问的节点。我们从根节点开始入栈，然后不断弹出栈顶元素进行处理。 如果当前节点是叶子节点，我们将其删除，并记录下其值，并将其父节点弹出栈。如果其父节点的左右子节点都被删除了，那么我们也将其删除，并返回其父节点。如果当前节点不是叶子节点，我们则按照右子节点先入栈的顺序，依次将其右子节点和左子节点入栈。 最后，我们返回记录下来的删除顺序即可，同时原树已经被完全删除。如果需要输出删除顺序，可以将其转换为字符串形式，如下所示： ```python root = TreeNode(1, TreeNode(2, TreeNode(4), TreeNode(5)), TreeNode(3, TreeNode(6), TreeNode(7))) res = del_leaf_nodes(root) print("->".join(map(str, res))) # 输出：4->5->2->6->7->3->1 ```