给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
示例 :
给定二叉树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 // 加上根节点自身构成的一条直径边
// }