814. Binary Tree Pruning的C++解法

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/musechipin/article/details/82661006

从叶子节点开始向上处理，只要是0就都剪掉。对于一棵子树，分两种情况，根节点是1的话这个节点不会消失，只要处理左右子树即可。根节点是0的话要注意如果左右子树都是0则这棵树全部剪掉，否则只剪掉不存在的。速度打败100%。

class Solution {
public:
	TreeNode* pruneTree(TreeNode* root) {
		if (root == NULL) return NULL;
		if (root->val == 1)
		{
			root->left =pruneTree(root->left);
			root->right =pruneTree(root->right);
			return root;
		}
		else 
        {
            root->left =pruneTree(root->left);
			root->right =pruneTree(root->right);
            if ((root->left == NULL)&&(root->right == NULL))
                return NULL;
            else return root;
        }
	}
};