Diameter of Binary Tree

求树的直径,其实就是对于每个节点左右子树深度和的最大值。
例如:
1
左边深度加右边深度,可知为3。但要注意下面的情况:
这里写图片描述
这时有个子节点的深度和(8)和已经超过根节点深度和(7)。

=============================
想法是对于每个节点的左右子树深度相加,取其中的最大值:

class Solution {
public:
    int diameterOfBinaryTree(TreeNode* root) {
        if(!root)
            return 0;
        int leftH = 0, rightH = 0;
        if(root->left)
            leftH = getHeight(root->left);
        if(root->right)
            rightH = getHeight(root->right);
        return max((leftH + rightH), max(diameterOfBinaryTree(root->left), diameterOfBinaryTree(root->right)));
    }
private:
    int getHeight(TreeNode* root)
    {
        if(!root)
            return 0;
        int left = getHeight(root->left);
        int right = getHeight(root->right);
        return left > right ? (left + 1) : (right + 1);
    }
};

然而这样嵌套递归效率不高。
然后是别人家的代码:

public class Solution {
public:
     int diameterOfBinaryTree(TreeNode root) {
        maxDepth(root);
        return max;
    }
private:
     int max = 0;
     int maxDepth(TreeNode root) {
        if (root == null) return 0;

        int left = maxDepth(root.left);
        int right = maxDepth(root.right);

        max = max(max, left + right);//左右子树最大深度相加

        return max(left, right) + 1;//取最大深度
    }
}

思路差不多,取左子树的最大深度,取右子树的最大深度,两者相加。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值