Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Node:
- def __init__(self, name):
- self.parent = None
- # self.children = set()
- self.name = name
- self.orbits = -1
- def __str__(self):
- if self.parent:
- r = "( {} : {} ".format(self.name, self.parent.name)
- else:
- r = "( {} : {} ".format(self.name, "-")
- if self.orbits != -1:
- r += "@ {} ".format(self.orbits)
- r += ")"
- return r
- def set_parent(self, n):
- self.parent = n
- def get_orbits_to_root(self):
- if self.parent:
- return self.parent.get_orbits_to_root() + 1
- else:
- return 0
- __repr__ = __str__
Advertisement
Add Comment
Please, Sign In to add comment