每天一道Code 814. 二叉树剪枝

代码思路
第一反应应该就是递归解决。
当前节点需要删除的条件有三个：

1. 当前节点值为0
2. 左子树没有1
3. 右子树没有1

代码

/**
* 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 one(TreeNode *root){
       if(!root) return false;
       if(!one(root->left)) root->left = NULL;     //左子树没有1，则删除左子树
       if(!one(root->right)) root->right = NULL;   //右子树没有1，则删除右子树
       if(root->val) return true;                  //节点值为1，返回1
       else{                                       
           if(!root->left && !root->right) return false;  //节点值为0,且左右子树都被删除，则返回0
           else return true;                              //有子树没被删除则返回1
       }
   }
   TreeNode* pruneTree(TreeNode* root) {
       if(!root) return root;
       if(one(root)) return root;
       else return NULL;
   }
};