【二叉树】删除给定值的叶子节点

0x00 题目

给你一棵以 root 为根的二叉树和一个整数 target
请你删除所有值为 target 的 叶子节点

注意，一旦删除值为 target 的叶子节点
它的父节点就可能变成叶子节点
如果新叶子节点的值恰好也是 target
那么这个节点也应该被删除

也就是说，你需要重复此过程直到不能继续删除

---

0x01 思路

因为要删除叶子节点
所以要判断左右节点是否空
所以要使用后序遍历方式
当左右节点为空且等于目标值时
返回一个空节点
上层节点更新返回的节点即可

---

0x02 解法

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法：

func removeLeafNodes(_ root: TreeNode?, _ target: Int) -> TreeNode? {
    guard let root = root else { return nil }
    
    let left = removeLeafNodes(root.left, target)
    let right = removeLeafNodes(root.right, target)
    
    // 更新节点
    root.left = left
    root.right = right
    
    // 是否删除
    if left == nil && right == nil && root.val == target {
        return nil
    }
    
    return root
}

---

0x03 我的小作品

欢迎体验我的作品之一：小五笔
五笔学习好帮手~
App Store 搜索即可~

---