Самый низкий общий предок бинарного дерева - рекурсивное решение

Question

Это вопрос кода leetcode: https://leetcode.com/problems/lowest-common-ancestor-of-a-binary-tree/

Мой код здесь

class CommonAncestor:

    def traverse(self, root, A, B):

        if not root:
            return None

        return self.helper(root, A, B)

    def helper(self, node, A, B):
        print(node)
        if node:        

            left = self.helper(node.left, A, B)
            right = self.helper(node.right, A, B)

            if node.val == A or node.val == B:
                return node

            if left and right:
                return node
            elif left or right:
                return left or right
            else:
                return None

Может кто-нибудь объяснить, почему этот код не работает? Он отлично работает на моем коде VS, но не может пройти тестовый пример. Я создал точно такое же дерево здесь:

class TreeNode:

    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None


obj = CommonAncestor()

root = TreeNode(3)

root.left = TreeNode(5)
root.left.left = TreeNode(6)
root.left.right = TreeNode(2)
root.left.right.left = TreeNode(7)
root.left.right.right = TreeNode(4)

root.right = TreeNode(1)
root.right.left = TreeNode(0)
root.right.right = TreeNode(8)
root.right.right.left = None
root.right.right.right = None

Я не могу понять, является ли это указанием leetcode c или его моим кодом ... Заранее большое спасибо!

Answer 1

Хорошо, разобрался ... Тип ввода LeetCode был TreeNode, а мой тип ввода был int ...