在 Python 中找出给定二叉树中是否存在 BST 的程序

假设我们有一个二叉树。我们必须从二叉搜索树(BST)树中找出最大的子树。我们返回 BST 的根节点。

所以,如果输入像

那么输出将是 -

示例

让我们看看以下实现以更好地理解 -

class TreeNode:

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

     self.val= val

     self.left= left

     self.right= right

def insert(temp,data):

   que = []

   que.append(temp)

   while (len(que)):

      temp = que[0]

      que.pop(0)

      if (not temp.left):

         if data is not None:

           temp.left= TreeNode(data)

         else:

           temp.left= TreeNode(0)

         break

      else:

         que.append(temp.left)

      if (not temp.right):

         if data is not None:

           temp.right= TreeNode(data)

         else:

           temp.right= TreeNode(0)

         break

      else:

         que.append(temp.right)

def make_tree(elements):

   Tree= TreeNode(elements[0])

   for element in elements[1:]:

      insert(Tree, element)

   return Tree

def print_tree(root):

   if root is not None:

      print_tree(root.left)

      print(root.val, end = ', ')

      print_tree(root.right)

def solve(root):

   c, m = 0, None

   def recurse(node):

      if node:

         nonlocal c, m

         left_val = recurse(node.left)

         right_val = recurse(node.right)

         count = -float("inf")

         if (node.left == None or node.left.val <= node.val) and (node.right == None ornode.val<= node.right.val):

            count = left_val + right_val + 1

         if count > c:

            c = count

            m = node

         return count

      return 0

   recurse(root)

   return m

tree = make_tree([1, 4, 6, 3, 5])

print_tree(solve(tree))

输入

tree = make_tree([1, 4, 6, 3, 5])

print_tree(solve(tree))

输出结果
3, 4, 5,

以上是 在 Python 中找出给定二叉树中是否存在 BST 的程序 的全部内容, 来源链接: utcz.com/z/363258.html

回到顶部