LeetCode 814 二叉树剪枝[dfs] HERODING的LeetCode之路

Posted 2022-07-28 HERODING23

解题思路：
一道非常标准的dfs题目，对原节点不断往下直到底部，遇到空节点返回空，遇到叶子节点且值为0的返回空，按理说在剪枝过程中，如果当前节点为0，如果左右子节点都不存在或者值都为0才剪枝，但是在dfs过程中，值为0的叶子节点已经返回nullptr了，也就是每次剪枝的都是叶子节点了（不是叶子的也因为左右子节点的剪枝变成叶子），代码如下：

/**
 * 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) 
        if(root == nullptr) return nullptr;
        root->left = pruneTree(root->left);
        root->right = pruneTree(root->right);
        if(root->left == nullptr && root->right == nullptr && root->val == 0) 
            return nullptr;
        
        return root;
    
;