力扣993,二叉树的堂兄弟节点:
在二叉树中,根节点位于深度 0 处,每个深度为 k 的节点的子节点位于深度 k+1 处。 如果二叉树的两个节点深度相同,但父节点不同 ,则它们是一对堂兄弟节点。
我们给出了具有唯一值的二叉树的根节点 root ,以及树中两个不同节点的值 x 和 y 。 只有与值 x 和 y 对应的节点是堂兄弟节点时,才返回 true 。否则,返回 false。
只想到了愚蠢 bfs,且较为啰嗦:
- 若当前节点的孩子是 x or y,则记录该父节点;
- 记录当前节点是否为目标兄弟节点;
- 判断这俩目标兄弟节点的父节点是否相同;
func isCousins(root *TreeNode, x int, y int) bool {
queue := make([]*TreeNode, 0)
queue = append(queue, root)
xis, yis := false, false
xDad, yDad := root, root
for len(queue) != 0{
length := len(queue)
for i := 0; i < length; i++{
cur := queue[0]
queue = queue[1:]
if cur.Left != nil{
if cur.Left.Val == x{
xDad = cur
}
if cur.Left.Val == y{
yDad = cur
}
queue = append(queue, cur.Left)
}
if cur.Right != nil{
if cur.Right.Val == x{
xDad = cur
}
if cur.Right.Val == y{
yDad = cur
}
queue = append(queue, cur.Right)
}
if cur.Val == x{
xis = true
}
if cur.Val == y{
yis = true
}
if xis && yis && (xDad != yDad){
return true
}
}
xis = false
yis = false
}
return false
}