Advertisement
Guest User

Untitled

a guest
Jan 20th, 2019
67
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.94 KB | None | 0 0
  1. import random
  2.  
  3. class XmasTree():
  4.     def __init__(self,value=None,left=None,right=None):
  5.         self.value=value
  6.         self.left=left
  7.         self.right=right
  8.  
  9. def print_tree(tree,level=0,sep=''):
  10.     if tree == None: return
  11.     print_tree(tree.right,level+1,".--")
  12.     print '  '*(level-1)+sep+str(tree.value)
  13.     print_tree(tree.left,level+1,"'--")
  14.  
  15. def build_tree(decorations, size = 5):
  16.     def build_tree_helper(decorations, counter):
  17.         if counter < size:
  18.             node = XmasTree(random.choice(decorations), build_tree_helper(decorations, counter+1), build_tree_helper(decorations, counter+1))
  19.             return node
  20.     return build_tree_helper(decorations, 0)
  21.  
  22. def is_xmas_tree(tree):
  23.     if tree==None:
  24.         return True
  25.     if abs(n_nodes(tree.left)-n_nodes(tree.right))>1:
  26.         return False
  27.     return is_xmas_tree(tree.left) + is_xmas_tree(tree.right)
  28.  
  29. def n_nodes(tree):
  30.     if tree == None:
  31.         return 0
  32.     return n_nodes(tree.left) + n_nodes(tree.right) + 1
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement