LeetCode 543. 二叉树的直径

543. 二叉树的直径

  

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

示例 :
给定二叉树

          1
         / \
        2   3
       / \     
      4   5    

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

注意：两结点之间的路径长度是以它们之间边的数目表示。

思路：参考题解：LeetCode题解

时间复杂度：O(N)，N 为树的节点数量，计算树的深度需要遍历所有节点。

空间复杂度：O(Height)，其中 Height 为二叉树的高度。由于递归函数在递归过程中需要为每一层递归函数分配栈空间，所以这里需要额外的空间且该空间取决于递归的深度，而递归的深度显然为二叉树的高度，并且每次递归调用的函数里又只用了常数个变量，所以空间复杂度为O(Height) 。

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */

// 参考题解：https://leetcode.cn/problems/diameter-of-binary-tree/solutions/557966/go-yu-yan-shi-yong-quan-ju-bian-liang-la-2crn/?languageTags=golang
// 【前序遍历】思想：
// 任意一条路径均可以被看作由某个节点为起点，从其左儿子和右儿子向下遍历的路径拼接得到。
func diameterOfBinaryTree(root *TreeNode) int {
    res := 0 // 注意：每次都要初始化res=0 ,不同用例在跑的时候 max会被覆盖导致 max值为上次的值
    
    var dfs func(node *TreeNode) int 
    dfs = func(node *TreeNode) int {
        if node == nil {
            return 0
        }
    
        l := dfs(node.Left)
        r := dfs(node.Right)
        res = max(res, l + r) // 注意：计算当前最大直径时，也可以仅计算直径边长数l+r，不需加1

        return max(l, r) + 1  // 加上根节点自身构成的一条直径边
    }

    dfs(root)

    return res
}

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

    return b
}

// var res int
// func diameterOfBinaryTree(root *TreeNode) int {
//     res = 0
//     dfs(root)

//     return res - 1
// }

// func dfs(node *TreeNode) int {
//     if node == nil {
//         return 0
//     }

//     // 错误：最后return时已经加了1了，因此这里不用return 1
//     // 遍历到叶子节点，叶子节点自身不占 “边长”
//     // if node.Left == nil && node.Right == nil {
//     //     return 1
//     // }
 
//     l := dfs(node.Left)
//     r := dfs(node.Right)
//     res = max(res, l + r + 1)

//     return max(l, r) + 1 // 加上根节点自身构成的一条直径边
// }