二叉树剪枝（leetcode 814）

二叉树子树移除算法

最新推荐文章于 2022-09-16 22:54:22 发布

原创最新推荐文章于 2022-09-16 22:54:22 发布 · 362 阅读

0 ·

CC 4.0 BY-SA版权

面试准备算法专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种算法，用于移除二叉树中所有不包含1的子树，通过对二叉树进行递归处理，实现了对特定条件子树的有效移除。

题面：

给定二叉树根结点 root ，此外树的每个结点的值要么是 0，要么是 1。

返回移除了所有不包含 1 的子树的原二叉树。

public TreeNode pruneTree(TreeNode root) {
        if (root != null){
            root.left = pruneTree(root.left);
            root.right = pruneTree(root.right);
            if (root.left == null && root.right == null && root.val == 0) {
                root = null;
            }
        }
        return root;
    }