Go算法练习—二叉树的直径

二叉树的直径

---

题目描述：

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

示例 :
给定二叉树

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

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

题目详解：

package main

import "fmt"

type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left  *TreeNode
 *     Right *TreeNode
 * }
 */
func diameterOfBinaryTree(root *TreeNode) int {
	if root == nil {
		return 0
	}
	diameter := 0
	depth(root, &diameter)
	return diameter
}

func depth(node *TreeNode, max *int) int {
	if node == nil {
		return 0
	}
	leftHeight := depth(node.Left, max)
	rightHeight := depth(node.Right, max)
	*max = myMax(*max, leftHeight+rightHeight)
	return myMax(leftHeight, rightHeight) + 1
}

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

func main() {
	var root = new(TreeNode)
	root.Val = 3
	fmt.Println(diameterOfBinaryTree(root))
}