1325. 删除给定值的叶子节点

package tree;


/**
 * 1325. 删除给定值的叶子节点
 * 给你一棵以 root 为根的二叉树和一个整数 target ，请你删除所有值为 target 的 叶子节点 。
 *
 * 注意，一旦删除值为 target 的叶子节点，它的父节点就可能变成叶子节点；如果新叶子节点的值恰好也是 target ，那么这个节点也应该被删除。
 *
 * 也就是说，你需要重复此过程直到不能继续删除。
 *
 *
 *
 * @author  Tang
 * @date 2021/8/4
 */
public class RemoveLeafNodes {

    /**
     * 后序遍历
     *
     * 每个节点判断
     * 如果左节点 右节点都为空 说明是叶子节点
     * 如果val等于target，就返回null  告诉父节点自己为空了
     */
    public TreeNode removeLeafNodes(TreeNode root, int target) {
        if (root == null) {
            return null;
        }
        TreeNode left = removeLeafNodes(root.left, target);
        TreeNode right = removeLeafNodes(root.right, target);
        if (left == null && right == null && root.val == target) {
            return null;
        }
        root.left = left;
        root.right = right;
        return root;
    }


    public static void main(String[] args) {

    }

}