入门力扣自学笔记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.根节点判断是否删除或者保留

---

代码：

/**
 * 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:
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;
    }
};