二叉树的直径——搜索思想(DFS)

55 阅读1分钟

DFS深度优先搜索

深度优先搜索算法是一种用于遍历或搜索树或图的算法。这个算法会尽可能深地搜索树的分支。当节点v的所在边都己被探寻过,搜索将回溯到发现节点v的那条边的起始节点。这一过程一直进行到已发现从源节点可达的所有节点为止。如果还存在未被发现的节点,则选择其中一个作为源节点并重复以上过程,整个进程反复进行直到所有节点都被访问为止

Question1

image.png

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

func diameterOfBinaryTree(root *TreeNode) int {
	ans := 1
	var depth func(root *TreeNode) int
	depth = func(root *TreeNode) int {
		if root == nil {
			return 0
		}
		l, r := depth(root.Left), depth(root.Right)
		ans = max(ans, l+r+1)
		return max(l, r) + 1
	}
	depth(root)
	return ans - 1
}

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