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