【算法】递归+深搜: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;
}
};