Go语言：LeetCode-二叉树的直径

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

返回3，它的长度是路径[4,2,1,3]或者[5,2,1,3]。
注意：两结点之间的路径长度是以它们之间边的数目表示。

分析：题目给定的示例容易让人产生这样的一个错误想法：只要求出树的左右子树的高度之和即为树的最大直径，其实不然，如以下情况：

上图二叉树的直径不是5，而应该是6.这条直径的路径为[8,6,4,3,5,7,9]，它没有经过树根。因此正确的想法应该是：计算每一个结点的左右子树高度之和，然后取最大值作为该二叉树的直径。代码如下：

//getDepth函数用于计算树的高度
func getDepth(root *TreeNode) int {
	if root != nil {
		leftDepth := getDepth(root.Left)
		rightDepth := getDepth(root.Right)
		if leftDepth > rightDepth {
			return leftDepth + 1
		}
		return rightDepth + 1
	}
	return 0
}
var maxDiameter int //全局变量，最长直径值存储在此变量中
//PreTraverse函数是前序遍历整颗树，对每一个结点测量其左右子树的高度
func PreTraverse(root *TreeNode) {
	if root != nil {
		leftDepth := getDepth(root.Left)
		rightDepth := getDepth(root.Right)
		if leftDepth+rightDepth > maxDiameter {
			maxDiameter = leftDepth + rightDepth  //更新最大直径值
		}
		PreTraverse(root.Left)
		PreTraverse(root.Right)
	}
}
func diameterOfBinaryTree(root *TreeNode) int {
	maxDiameter = 0
	PreTraverse(root)
	return maxDiameter
}