lintcode1181-二叉树的直径

给定一棵二叉树,计算树的直径长度,即任意两个节点之间最长路径。提供了一个解决方案,通过广度优先搜索遍历整棵树,计算每个节点的左右子树高度并寻找最大值。

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

   Question:
   给定一颗二叉树,您需要计算树的直径长度。 二叉树的直径是树中任意两个节点之间最长路径的长度。
   此路径不一定会通过树根。
 
   样例:
   给定一棵二叉树
            1
           / \
          2   3
         / \
        4   5
   输出:3
   因为,这是路径[4,2,1,3] 或者 [5,2,1,3]的长度.

   Solution:
   思路:本题的根本在于考察数的高度
   对于每个节点,例如根节点,计算左子树的高度,计算右子树的高度,相加,就是经过根节点的路径长度
   计算所有节点的左右子树的高度和
   找到最大值即可
 
   步骤:
   1、遍历整颗树(借助队列,广度优先搜索)
   2、对于每个节点,计算左右子树的高度,相加,如果比初始值max大,替换max为左右子树和
 

public class Q1181_diameterOfBinaryTree {

    public int diameterOfBinaryTree(TreeNode root) {
        if(root == null)
            return 0;
        int max = 0;
        //自顶向下遍历二叉树所有节点,借助队列
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(root);
        while(!queue.isEmpty()) {
            TreeNode temp = queue.poll();
            //左子树高度
            int left = height(temp.left);
            //右子树高度
            int right = height(temp.right);
            //找到最大的max
            if(left + right > max)
                max = left + right;
            if(temp.left != null)
                queue.offer(temp.left);
            if(temp.right != null)
                queue.offer(temp.right);
        }
        return max;
    }

    /**
     * 计算一个树的高度
     * @param node
     * @return
     */
    public int height(TreeNode node) {
        if(node == null)
            return 0;
        int leftHeight = height(node.left);
        int rightHeight = height(node.right);
        return Math.max(leftHeight,rightHeight) + 1;
    }

    @Test
    public void test() {
        TreeNode t = new TreeNode(1);
        t.left = new TreeNode(2);
        t.right = new TreeNode(3);
        t.left.left = new TreeNode(4);
        t.left.right = new TreeNode(5);
        System.out.println(diameterOfBinaryTree(t));
    }
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值