[树] leetcode 814 Binary Tree Pruning

发布时间：2020-12-14 05:11:36 所属栏目：大数据来源：网络整理

导读：problem:https://leetcode.com/problems/binary-tree-pruning/ ? ? ? ? 一道非常简单的tree题目，根据题意模拟剔除不包含0的子树即可。 class Solution { public : bool dfs(TreeNode* p) { if (!p) return false ; bool bLeft = dfs(p- left); bool bRight

problem:https://leetcode.com/problems/binary-tree-pruning/

? ? ? ? 一道非常简单的tree题目，根据题意模拟剔除不包含0的子树即可。

class Solution {
public:
    bool dfs(TreeNode* p)
    {
        if (!p) return false;
        bool bLeft = dfs(p->left);
        bool bRight = dfs(p->right);
        
        if (!bLeft)
        {
            p->left = nullptr;
        }
        if (!bRight)
        {
            p->right = nullptr;
        }
        if (bLeft || bRight || p->val == 1)
        {
            return true;
        }

        return false;    

    }
    TreeNode* pruneTree(TreeNode* root) {
        dfs(root);
        return root;
    }
};

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!