LeetCode 543. 二叉树的直径 动态规划树形DP详解

543. 二叉树的直径

题目来源

543. 二叉树的直径

题目分析

给定一棵二叉树,要求计算其直径长度。二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能经过根结点。

题目难度

  • 难度:简单

题目标签

  • 标签:动态规划, 数组

题目限制

  • 1 <= 树的结点个数 <= 10000

解题思路

思路1:递归 + 深度优先搜索 (DFS)

  1. 定义问题

    • 树的直径长度是任意两个结点之间路径的最大值,可以认为等于以某个结点为根的左子树深度和右子树深度之和。
  2. 状态转移

    • 采用 DFS 的递归方式,从叶节点开始逐层返回,计算每个节点的左子树深度和右子树深度,并更新最大直径。
  3. 递归公式

    • dfs(root) = max(dfs(root.left), dfs(root.right)) + 1
    • ans = max(ans, left + right),其中 leftright 分别为左子树和右子树的深度。
  4. 最终结果

    • ans 即为所求的二叉树的最大直径。

核心算法步骤

  1. DFS 递归

    • 从根节点开始,通过 DFS 遍历整棵树,在遍历过程中更新当前节点的左子树和右子树的深度,并计算该节点作为根节点时的最大直径。
  2. 返回最大深度

    • 每次递归返回当前节点的最大深度,供上层递归调用。

代码实现

以下是求解二叉树直径的 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) 函数是一个递归函数,用于计算以当前节点为根的子树的深度,并更新树的直径。
  • leftright 分别表示当前节点左子树和右子树的深度。
  • 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 算法,我们可以有效地求解二叉树的直径问题。此方法思路清晰,易于实现,且具有较高的效率。


  • 3
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值