LeetCode—236. 二叉树的最近公共祖先

236. 二叉树的最近公共祖先

题目描述：
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

考察重点：DFS搜索两个节点，当找到目标节点时返回1，表示当前分支存在目标节点。回溯时，当左右子树都返回1，则认为是最近公共祖先

func findK1(res, root, p, q *TreeNode) int {
	var t1, t2 int
	if root == nil { //搜索到叶节点，返回0
		return 0
	}
	if root == p || root == q { //自己就是搜索的节点，则还需要判断子节点中还有无另一个搜索节点，如果有，直接程序终止，返回自己，否则返回1，表示这个分支有一个搜索节点
		if findK1(res, root.Left, p, q) == 1 || findK1(res, root.Right, p, q) == 1 {
			*res = *root
		}
		return 1
	}
	t1 = findK1(res, root.Left, p, q)
	t2 = findK1(res, root.Right, p, q)
	if t1 == 1 && t2 == 1 { //左右子树都有搜索节点的第一个节点，证明自己就是他们的最近父节点（直接wait.Done()返回自己）
		*res = *root
	} else if t1 == 1 || t2 == 1 { //左右子树一边有搜索节点，返回1，证明该分支有一个搜索节点
		return 1
	}
	return 0 //返回0表示这个分支无搜索节点
}
func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {
	var res TreeNode

	
	findK1(&res, root, p, q)

	return &res
}