python - 验证是否存在一行矩阵 M 其值属于树 T

Question

我有个问题。我需要构建这个函数来验证是否存在一行M其值属于 BST tree。我有一个由以下定义的树类：

class TreeNode(object):
    def __init__(self, value):
        self.left = None
        self.right = None
        self.data = value

我怎样才能做到这一点？

我的代码：

def checkRowOnTree(tree, M):
    if tree is None:
        return False
    n = len(M)
    m = len(M[0])
    flag = []
    for i in range(n):
        for j in range(m):
            if tree is not None:
                return(tree.data == (M[i][j])
                       and checkRowOnTree(tree.left, M)
                       and checkRowOnTree(tree.right, M))