【Leetcode_总结】993. 二叉树的堂兄弟节点 - python

最新推荐文章于 2021-05-18 20:30:11 发布

Maka_uir

最新推荐文章于 2021-05-18 20:30:11 发布

阅读量401

点赞数

分类专栏： Leetcode 文章标签：二叉树

本文链接：https://blog.csdn.net/maka_uir/article/details/87860364

版权

Leetcode 专栏收录该内容

195 篇文章 2 订阅

订阅专栏

本文介绍LeetCode中的第993题，探讨如何找到二叉树中深度相同但父节点不同的堂兄弟节点。通过Python实现算法，详细解释解题思路和代码细节。

摘要由CSDN通过智能技术生成

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

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

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

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

示例 1：

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

链接:https://leetcode-cn.com/problems/cousins-in-binary-tree/description/
思路：对二叉树进行层次遍历，如果一层包含两个给定节点值，且两个节点不属于同一个父节点，则输出True 否则是False
代码：

# 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':
        que = [(0,root)]
        while que:
            tempList = []
            for i in range(len(que)):
                node_ = que.pop(0)
                node = node_[1]
                tmp = node_[0]
                if node.val == x or node.val == y:
                    tempList.append((tmp,node.val))
                if node.left:
                    que.append((node.val,node.left))
                if node.right:
                    que.append((node.val,node.right))
            if len(tempList) == 2 and tempList[0][0] != tempList[1][0]:
                return True
        return False