Leetcode543. 二叉树的直径(HOT100)

kitesxian2024-12-05 8:34

代码：

cpp 复制代码

class Solution {
public:
    int res = 0;
    int diameterOfBinaryTree(TreeNode* root) {
        dfs(root);
        return res;
    }
    int dfs(TreeNode* root){
        if(!root)return 0;
        int left = dfs(root->left),right = dfs(root->right);
        res = max(res,left+right);
        return max(left,right)+1;
    }
};

题解：

要求最大直径，其实有点类似于深度，以root节点为例，直径最大就是左子树的3+右子树的4，答案就是7。

对于二叉树来说，我们使用dfs遍历所有点，自下而上，把子节点的值返回给父节点，最终算出整个树的最大直径。

还有一个注意的点：给上一层返回的时候应该返回多少？当然是你往左走（left的值）和往右走（right）的值的最大值，然后再加上1--------上一层来到当前节点，也需要走一步。