Diameter of Binary Tree

来源:互联网 发布:淘宝店被永久封店 编辑:程序博客网 时间:2024/06/05 15:25

求树的直径,其实就是对于每个节点左右子树深度和的最大值。
例如:
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;//取最大深度    }}

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

原创粉丝点击