LeetCode 993 - 二叉树的堂兄弟节点

class Solution:
    def isCousins(self, root: Optional[TreeNode], x: int, y: int) -> bool:
        # 记录值进哈希表里
        targets = {x, y}
        # 结果返回存储
        parrentNodeAndDepth = []

        # 进入遍历前要先保证根不为空且根节点值不在哈希表里
        if not root or root.val in targets:
            return False
        
        def isCousinsIMPL(parrent: Optional[TreeNode], root: Optional[TreeNode], depth: int):
            nonlocal parrentNodeAndDepth

            # 当前节点如果为空则返回
            if not root:
                return
            # 如果存在，则记录父节点值和当前节点深度
            if root.val in targets:
                parrentNodeAndDepth.append([depth, parrent.val])
            
            # 递归时，深度+1，同时把父节点向下传递
            isCousinsIMPL(root, root.left, depth + 1)
            isCousinsIMPL(root, root.right, depth + 1)
                
        isCousinsIMPL(None, root, 0)
        itemX = parrentNodeAndDepth[0]
        itemY = parrentNodeAndDepth[1]
        # 同时满足父节点值不相同和节点深度相同
        return itemX[0] == itemY[0] and itemX[1] != itemY[1]