给定二叉树根结点 root ,此外树的每个结点的值要么是 0,要么是 1。
返回移除了所有不包含 1 的子树的原二叉树。 把只包含0的子树删除(断开)
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/binary-tree-pruning
class Solution {
public:
TreeNode* pruneTree(TreeNode* root) {
if(root == NULL)
return NULL;
root->left = pruneTree(root->left);
root->right = pruneTree(root->right);
if(!root->left && !root->right && root->val == 0)
return NULL;
else
return root;
}
};