Update and rename bst_all_subnodes.py to bst_all_subnodes.py

This commit is contained in:
bt3gl 2023-08-08 13:28:30 -07:00 committed by GitHub
parent cd5ac00126
commit eedb6b6863
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,39 +0,0 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# author: bt3gl
class Node:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def all_possible_bst(start, end, memo):
result = []
if start > end:
result.append(None)
return result
if (start, end) in memo:
return memo[(start, end)]
for i in range(start, end + 1):
left = all_possible_bst(start, i - 1, memo)
right = all_possible_bst(i + 1, end, memo)
for l in left:
for r in right:
root = Node(i, l, r)
result.append(root)
memo[(start, end)] = result
return result
def generate_trees(n):
memo = {}
return all_possible_bst(1, n, memo)