Leetcode 814. 二叉树剪枝

最新推荐文章于 2020-03-05 16:54:11 发布

达达达达锅

最新推荐文章于 2020-03-05 16:54:11 发布

阅读量323

点赞数

分类专栏： Leetcode 文章标签： Leetcode

本文链接：https://blog.csdn.net/liaoxuda_edu/article/details/81565697

版权

Leetcode 专栏收录该内容

43 篇文章 0 订阅

订阅专栏

解题思路：

一句话：当一个节点的左（右）节点全是0，那么将这个节点的左（右）节点设置为NULL。若不是，那么将这个节点的左节点和右节点进行递归。

AC解：

class Solution {
public:
	TreeNode * pruneTree(TreeNode* root) {
		if (root == NULL)
			return NULL;
		Dele_Tree(root);
		return root;
	}
	void Dele_Tree(TreeNode* Now_root)
	{
		if (Now_root == NULL)
			return;
		if (IsZero(Now_root->left))
			Now_root->left = NULL;
		else
			Dele_Tree(Now_root->left);
		if (IsZero(Now_root->right))
			Now_root->right = NULL;
		else
			Dele_Tree(Now_root->right);
                return;
	}

	bool IsZero(TreeNode* Root)
	{
		if (Root == NULL)
			return true;
		if (Root->val == 1)
			return false;
		return (IsZero(Root->left) && IsZero(Root->right));
	}
};