剑指 Offer II 047. 二叉树剪枝

后序遍历

1. 从叶子节点一个一个删，先看左子树，再看右子树，然后再看当前节点

2. 假设叶子节点为零直接删除（由于递归，节点左右子树都为零的话，该节点也会称为叶子节点），否则不进行删除并返回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:
    TreeNode* pruneTree(TreeNode* root) {
        int res = prune(root);
        if(res){
            return root;
        }else{
            return NULL;
        }
    }
    int prune(TreeNode* root){
        int flag = 0;
        // 节点为空时返回0
        if(root==nullptr){
            return 0;
        }
        // 左子树可以剪掉
        if(prune(root->left) == 0){
            root->left = nullptr;
        }else{
            // 子树有不为0的值该节点也不能剪掉
            flag = 1;
        }
        // 右子树可以剪掉
        if(prune(root->right) ==0){
            root->right = nullptr;
        }else{
            flag = 1;
        }
        if(flag){
            return flag;
        }
        // 当左右子树都剪掉时，查看当前节点值
        if(root->val==1){
            return 1;
        }else{
            return 0;
        }
    }
};