栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

入门力扣自学笔记99 C++ (题目编号814)

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

入门力扣自学笔记99 C++ (题目编号814)

814. 二叉树剪枝

题目:

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

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

节点 node 的子树为 node 本身加上所有 node 的后代。


示例 1:

输入:root = [1,null,0,0,1]
输出:[1,null,0,null,1]
解释:
只有红色节点满足条件“所有不包含 1 的子树”。 右图为返回的答案。


示例 2:

输入:root = [1,0,1,0,0,0,1]
输出:[1,null,1,null,1]


示例 3:

输入:root = [1,1,0,1,1,0,1,0]
输出:[1,1,0,1,1,null,1]


提示:

树中节点的数目在范围 [1, 200] 内
Node.val 为 0 或 1


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


思路:

递归删除根节点。
1.判断是否删除该节点的条件是3个:
(1)自身是否为0
(2)左子节点是否可以删除,递归
(3)右子节点是否可以删除,递归

2.使用“或”融合步骤1中的三个条件,作为当前节点是否要删除
3.根节点判断是否删除或者保留


代码:
class Solution {
public:
bool dfsDel(TreeNode* node){
    //递归终止条件
    if(node == NULL){
        return false;
    }
    
    //自身节点判断条件
    bool bs;
    if(node->val == 0){
        bs = false;
    }else{
        bs = true;
    }

    //左子节点和右子节点判断条件
    bool bl = dfsDel(node->left);
    bool br = dfsDel(node->right);

    //删左子节点
    if(bl == false){
        node->left = NULL;
    }

    //删右子节点
    if(br == false){
        node->right = NULL;
    }

    //返回最终判断条件
    return bs||bl||br;
}

TreeNode* pruneTree(TreeNode* root) {
      bool res = dfsDel(root);
      if(res)
        return root;
      else
        return NULL;
    }
};
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1022522.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号