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

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​​ 搜索即可~