LintCode 921. 统计唯一值子树个数

最新推荐文章于 2024-11-27 09:17:01 发布

yeshichen

最新推荐文章于 2024-11-27 09:17:01 发布

阅读量341

点赞数

CC 4.0 BY-SA版权

分类专栏： LintCode and LeetCode

本文链接：https://blog.youkuaiyun.com/xiongxiong966/article/details/89431205

LintCode and LeetCode 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种算法，用于统计给定二叉树中具有相同节点值的唯一子树的数量。通过深度优先搜索(DFS)遍历二叉树，检查每个子树是否满足唯一值条件，最终返回所有符合条件的子树总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述
给定一棵二叉树，统计唯一值子树的数目.
唯一值子树意味着子树的所有节点都具有相同的值.

样例

输入: root = {5,1,5,5,5,#,5}
输出: 4
解释:
              5
             / \
            1   5
           / \   \
          5   5   5



输入: root = {1,3,2,4,5,#,6}
输出: 3
解释:
          1
         / \
        3   2
       / \   \
      4   5   6

分析：

1.一个root下面要为唯一值子树的话，必须满足left和right都是唯一值子树，并且，left和right必须和root相等
2.dfs的时候每次判定一个非null节点为true的话，count++，最后count就是所有唯一子树的个数

JAVA代码：

private int res;

public int countUnivalSubtrees(TreeNode root) {
	dfs(root);
	return res;
}

private boolean dfs(TreeNode root) {
	if (root.left == null && root.right == null) {
		res++;
		return true;
	}
	if (root.left != null && root.right != null) {
		boolean flag1 = dfs(root.left), flag2 = dfs(root.right);
		if (flag1 && flag2 && root.val == root.left.val && root.val == root.right.val) {
			res++;
			return true;
		}
	} else if (root.left == null) {
		if (dfs(root.right) && root.val == root.right.val) {
			res++;
			return true;
		}
	} else if (root.right == null) {
		if (dfs(root.left) && root.val == root.left.val) {
			res++;
			return true;
		}
	}
	return false;
}