二叉树的直径

题目描述：

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

示例 :
给定二叉树

      1
     / \
    2   3
   / \     
  4   5    

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

思路:

Dfs 递归判断，每一次都需要判断最大直径是否比已经找寻的最大直径大

代码:

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func diameterOfBinaryTree(root *TreeNode) int {
    var dfs func(root *TreeNode)int
    ans := 0
    dfs = func(root *TreeNode)int{
        if  root == nil{
            return 0
        }
        L := dfs(root.Left)
        R :=dfs(root.Right)
        ans = max(ans, L+R)
        return max(L,R) +1
    }
    dfs(root)
    return ans

}

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

代码效率：

执行用时：4 ms, 在所有 Go 提交中击败了91.28%的用户
内存消耗：4.2 MB, 在所有 Go 提交中击败了92.36%的用户