543. 二叉树的直径
题目来源
题目分析
给定一棵二叉树,要求计算其直径长度。二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能经过根结点。
题目难度
- 难度:简单
题目标签
- 标签:动态规划, 数组
题目限制
1 <= 树的结点个数 <= 10000
解题思路
思路1:递归 + 深度优先搜索 (DFS)
-
定义问题:
- 树的直径长度是任意两个结点之间路径的最大值,可以认为等于以某个结点为根的左子树深度和右子树深度之和。
-
状态转移:
- 采用 DFS 的递归方式,从叶节点开始逐层返回,计算每个节点的左子树深度和右子树深度,并更新最大直径。
-
递归公式:
dfs(root) = max(dfs(root.left), dfs(root.right)) + 1
ans = max(ans, left + right)
,其中left
和right
分别为左子树和右子树的深度。
-
最终结果:
ans
即为所求的二叉树的最大直径。
核心算法步骤
-
DFS 递归:
- 从根节点开始,通过 DFS 遍历整棵树,在遍历过程中更新当前节点的左子树和右子树的深度,并计算该节点作为根节点时的最大直径。
-
返回最大深度:
- 每次递归返回当前节点的最大深度,供上层递归调用。
代码实现
以下是求解二叉树直径的 Java 代码:
/**
* 543. 二叉树的直径
* @param root 二叉树
* @return 二叉树的直径长度
*/
public int diameterOfBinaryTree(TreeNode<Integer> root) {
dfs(root);
return ans;
}
private int ans = 0;
private int dfs(TreeNode<Integer> root) {
if (root == null) {
return -1;
}
int left = dfs(root.left) + 1;
int right = dfs(root.right) + 1;
ans = Math.max(ans, left + right);
return Math.max(left, right);
}
代码解读
dfs(root)
函数是一个递归函数,用于计算以当前节点为根的子树的深度,并更新树的直径。left
和right
分别表示当前节点左子树和右子树的深度。ans
用于存储二叉树的最大直径,每次递归过程中更新最大值。
性能分析
- 时间复杂度:
O(N)
,其中N
是树的节点数量。每个节点遍历一次。 - 空间复杂度:
O(H)
,其中H
是树的高度。递归栈的空间开销。
测试用例
以下是一些测试用例,用于验证代码的正确性:
TreeNode<Integer> root1 = new TreeNode<>(1);
root1.left = new TreeNode<>(2);
root1.right = new TreeNode<>(3);
root1.left.left = new TreeNode<>(4);
root1.left.right = new TreeNode<>(5);
System.out.println(diameterOfBinaryTree(root1)); // 输出: 3
TreeNode<Integer> root2 = new TreeNode<>(1);
root2.left = new TreeNode<>(2);
root2.left.left = new TreeNode<>(3);
System.out.println(diameterOfBinaryTree(root2)); // 输出: 2
TreeNode<Integer> root3 = new TreeNode<>(1);
root3.right = new TreeNode<>(2);
root3.right.right = new TreeNode<>(3);
System.out.println(diameterOfBinaryTree(root3)); // 输出: 2
扩展讨论
总结
通过递归和 DFS 算法,我们可以有效地求解二叉树的直径问题。此方法思路清晰,易于实现,且具有较高的效率。