开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第26天，点击查看活动详情

一、题目描述：

993. 二叉树的堂兄弟节点 - 力扣（LeetCode） (leetcode-cn.com)

在二叉树中，根节点位于深度 0 处，每个深度为 k 的节点的子节点位于深度 k+1 处。

如果二叉树的两个节点深度相同，但 父节点不同 ，则它们是一对堂兄弟节点。

我们给出了具有唯一值的二叉树的根节点 root ，以及树中两个不同节点的值 x 和 y 。

只有与值 x 和 y 对应的节点是堂兄弟节点时，才返回 true 。否则，返回 false。

示例 1：

输入：root = [1,2,3,4], x = 4, y = 3
输出：false

示例 2：

输入：root = [1,2,3,null,4,null,5], x = 5, y = 4
输出：true

示例 3：

输入：root = [1,2,3,null,4], x = 2, y = 3
输出：false

提示：

二叉树的节点数介于 2 到 100 之间。
每个节点的值都是唯一的、范围为 1 到 100 的整数。

二、思路分析：

将每个节点的父结点，每个节点的层数和每个节点的索引放进队列中；
在遍历队列的时候，构建每个节点的字典，将每个节点的值作为key，每个节点的父结点和所在层所作为value；
得到所有数据的字典信息，查询X和Y的信息是否匹配，输出最后的结果；
为了节省时间，当查询到x和y之后，就结束队列遍历；

三、AC 代码：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isCousins(self, root: TreeNode, x: int, y: int) -> bool:
        if not root:
            return False

        dicts = {}   # 用于保存每个节点的信息（每个节点所在的层数和父结点）
        queue = collections.deque()   # 创建一个队列
        queue.append([root,None,1])   # 将根节点放进队列中
        flag_x = False   # 用于标记是否已经遍历到x
        flag_y = False   # 用于标记是否已经遍历到y
        while queue:
            roots,froot,d = queue.popleft();  # 获得当前节点的父结点，当前节点的层数
            dicts[roots.val] = [froot,d]   # 将当前节点的信息放到字典中

            if roots.val == x:  # 如果x和y已经被搜索到，就结束队列遍历
                flag_x = True
            if roots.val == y:
                flag_y = True
            if flag_x and flag_y:
                break

            if roots.left:    # 如果左子节点不为空，则将左子节点的信息作为一个列表放进队列中
                queue.append([roots.left,roots,d+1])
            if roots.right:   # 如果右子节点不为空，则将右子节点的信息作为一个列表放进队列中
                queue.append([roots.right,roots,d+1])

        if dicts[x][0]!=dicts[y][0] and dicts[x][1]==dicts[y][1]:  # 查询字典中x和y的信息是否匹配
            return True
        return False

刷题笔记-993. 二叉树的堂兄弟节点

一、题目描述：

二、思路分析：

三、AC 代码：