力扣实战814题——二叉树剪枝

二叉树剪枝这道题我个人觉得比较简单，考的其实是二叉树的遍历方式，即通过递归的方式访问二叉树的节点，重点是这个树剪枝之后不能有只包含节点值为0的子树，我们可以转化一下思路，遇到有节点值为0且无左右子树的情况相当于满足条件了，就把这个节点值设为空，相当于剪掉了这个节点，所以我们利用递归，分别剪掉每一个节点的左右子树，即可，最后我们剪掉之后这个根节点也会变成那种光秃秃的0节点，所以要把这个节点也剪去，所以最后应该是加一条判断剪去这个节点，返回最后的节点值。所以给出我的代码：

 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* pruneTree(TreeNode* root) {
        if(root==NULL){
            return NULL;
        }
        else if(root->val==0&&root->left==NULL&&root->right==NULL){
            return NULL;
        }
        else{
            root->left=pruneTree(root->left);
            root->right=pruneTree(root->right);
            if(root->val==0&&root->left==NULL&&root->right==NULL){
            root=NULL;
        }
            return root;
        }
    }
};