自学内容网 自学内容网

【算法】递归+深搜:814.二叉树剪枝

目录

1、题目链接

2、题目

3、解法(后序遍历)

4、代码


1、题目链接

814.二叉树剪枝(LeetCode)

2、题目

3、解法(后序遍历)
 

 我们这次不使用宏观的观察法,而是从具体实现开始。

题目要求我们,去掉不含1的子树。

对于子树这个概念,如何判断是否不含1?

我们就需要先判断他的左右子树,然后再判断根节点。

也就是,我们需要自顶向上的遍历,因此,我们使用后序遍历,按照左子树、右子树、根节点的顺序。


剪枝操作的实施:把结点置空


遇到叶子节点,且val = 0,采取剪枝。

非叶子结点,直接返回对应的root.


函数的出口:

root == NULL

4、代码

class Solution {
public:

//后序遍历 
//因为想要剪掉一个子树,需要确定他的子节点是否需要剪枝
TreeNode* pruneTree(TreeNode* root) {
        if(root == NULL)
        return NULL;
        
    
    root->left = pruneTree(root->left);
    root->right = pruneTree(root->right);
    
    if(root->left== NULL && 
   root->right == NULL &&
   root->val == 0)
{
root = NULL;   
    }
    
    return root;
    }
};


原文地址:https://blog.csdn.net/m0_73726899/article/details/143581073

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!