Advertisement
Guest User

Untitled

a guest
Nov 12th, 2019
86
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.05 KB | None | 0 0
  1. class Tree:
  2.     def __init__(self, n):
  3.         self.root = n           # POCETNI BROJ
  4.         self.left  = None
  5.         self.right = None
  6.  
  7.     def __repr__(self):
  8.         ja = str(self.root) + '|'
  9.         if self.left:
  10.             ja += self.left.__repr__()
  11.         if self.right:
  12.             ja += self.right.__repr__()
  13.         ja += '||'
  14.         return ja
  15.  
  16.     def dfs(self, n):
  17.         if self.root == n:
  18.             exit(f'nasao sam {n}')
  19.         if self.left:
  20.             self.left.dfs(n)
  21.         if self.right:
  22.             self.right.dfs(n)
  23.  
  24.     def add(self, n):
  25.         if n == self.root:
  26.             return
  27.         if n < self.root and self.left == None:
  28.             self.left = Tree(n)
  29.         elif n > self.root and self.right == None:
  30.             self.right = Tree(n)
  31.         elif n < self.root:
  32.             self.left.add(n)
  33.         else:
  34.             self.right.add(n)
  35.  
  36. tree = Tree(12)
  37. tree.add(3)
  38. tree.add(14)
  39. tree.add(2)
  40. tree.add(5)
  41. tree.add(13)
  42. tree.add(17)
  43. tree.add(1)
  44. tree.add(20)
  45. tree.add(22)
  46.  
  47. tree.dfs(5)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement