剑指 Offer II 047. 二叉树剪枝-c语言-双百递归代码

最新推荐文章于 2023-03-17 15:03:20 发布

Mr Gao

最新推荐文章于 2023-03-17 15:03:20 发布

阅读量200

点赞数

分类专栏： c++与c语言力扣刷题文章标签： c语言算法

本文链接：https://blog.csdn.net/weixin_43327597/article/details/124470405

版权

c++与c语言同时被 2 个专栏收录

803 篇文章 5 订阅

订阅专栏

力扣刷题

787 篇文章 12 订阅

订阅专栏

剑指 Offer II 047. 二叉树剪枝-c语言-双百递归代码

给定一个二叉树根节点 root ，树的每个节点的值要么是 0，要么是 1。请剪除该二叉树中所有节点的值为 0 的子树。

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

示例 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]
解释:
在这里插入图片描述

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
 struct TreeNode*   f(struct TreeNode* root){
     if(root!=NULL){
        
         if(root->left){
             root->left=f(root->left);

         }
          if(root->right){
             root->right=f(root->right);

         }
          if(root->left==NULL&&root->right==NULL&&root->val==0){
             return NULL;
         }

     }
     return root;

 }


struct TreeNode* pruneTree(struct TreeNode* root){
    root=f(root);
    return root;

}