LeetCode-树-二叉树剪枝

原创于 2020-11-24 22:10:06 发布 · 148 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #算法 #剪枝 #leetcode

数据结构与算法同时被 2 个专栏收录

15 篇文章

订阅专栏

LeetCode

14 篇文章

订阅专栏

本文介绍如何通过树的后续遍历算法，删除二叉树中所有不包含1的子树，只保留含有1的节点，以简化表示。通过递归实现剪枝，示例和代码展示了操作过程和结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

( 节点 X 的子树为 X 本身，以及所有 X 的后代。)

示例1:
输入: [1,null,0,0,1]
输出: [1,null,0,null,1]

解释:
只有红色节点满足条件“所有不包含 1 的子树”。
右图为返回的答案。

示例2:
输入: [1,0,1,0,0,0,1]
输出: [1,null,1,null,1]

示例3:
输入: [1,1,0,1,1,0,1,0]
输出: [1,1,0,1,1,null,1]

说明:

给定的二叉树最多有 100 个节点。
每个节点的值只会为 0 或 1 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-pruning
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

树的后续遍历，

先判断左子树与右子树是不是全都为0，

如果都是，则判断当前节点的值是不是0，如果是，则将该节点置为nullptr

否则不进行任何操作

一直递归到叶子节点，然后回溯到根节点

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* pruneTree(TreeNode* root) {
        prunch(root);
        return root;
    }

    //注意此处传入的指针的引用，能够返回原来的值
    bool prunch(TreeNode*& node)
    {
        if(node == nullptr)
        {
            return true;
        }

        //对左子树剪枝，返回标志位
        bool bLeft = prunch(node->left);
        //对右子树剪枝，返回标志位
        bool bRight= prunch(node->right);
        //判断当前节点是否应该剪枝，并返回标志位
        if(bLeft && bRight && node->val == 0)
        {
            //delete node;
            node = nullptr;
            return true;
        }
        else
        {
            return false;
        }
    }
};