二叉树的直径

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

func diameterOfBinaryTree(root *TreeNode) int {
    maxDia := 0
    if root == nil{
        return maxDia
    }
    var dfs func(*TreeNode)int
    dfs = func(node *TreeNode)int{
        if node == nil{
            return 0
        }
        lh := dfs(node.Left)
        rh := dfs(node.Right)
        maxDia = max(maxDia, lh+rh)
        return 1 + max(lh, rh)
    }
    dfs(root)
    return maxDia
}

func max(a, b int) int {
    if a > b{
        return a
    }
    return b
}

我自己写的跟这个差不多，但是死活过不了 leetcode 的编译器。

思路是，先找到以一个节点为根节点的树的直径，再遍历所有点，求最大直径。