【LeetCode】二叉树直径(dfs)

力扣 二叉树的直径(dfs)

题目链接：https://leetcode-cn.com/problems/diameter-of-binary-tree/

题目的意思是求二叉树的直径，就是从任意节点出发，到任意节点结束，最长的那一条路径，也就是这条路上经过节点的数量减去1就是最长直径

需要明确的是，其最长直径不一定非得要经过根节点，所以我们应该以每个节点当做根节点，保存其路径最大值

同时，以该节点为根节点的最大路径即为max(左子树最大长度，右子树最大长度)+1

以该节点为根节点的长度为左子树长度+右子树长度

流程：

• 每个点为中心的直径就是这个点左右子树的深度之和；
• res保存最大直径，最大直径即为每次访问某个结点时获得的以该节点为中心的直径；
• 返回值为以该节点为根的子树的深度；

func max(a,b int)int{
	if a>b{
		return a
	}
	return b
}
var ans int
func dfs(root *TreeNode) int{
	if root==nil{
		return 0
	}
	x1:=dfs(root.Left)
	x2:=dfs(root.Right)
	result:=x1+x2 // 左子树长度+右子树长度 就等于以该节点为根节点的最大长度

	// 保存最大长度
	if result>ans{
		ans=result
	}

	// 以该节点为根节点的最大路径即为max(左子树最大长度，右子树最大长度)+1
	return max(x1,x2)+1
}
func diameterOfBinaryTree(root *TreeNode) int {
	ans=0
	dfs(root)
	return ans
}