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

解题思路：
一道非常标准的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;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23