| 202 | |
| 203 | |
| 204 | class TreeNode: |
| 205 | |
| 206 | def __init__(self, label, children=None, node_type='other'): |
| 207 | self.label = label |
| 208 | self.children = children if children is not None else [] |
| 209 | self.node_type = node_type |
| 210 | self.subtree_size = 0 |
| 211 | |
| 212 | def get_children(self): |
| 213 | return self.children |
| 214 | |
| 215 | def __str__(self): |
| 216 | return self.label |
| 217 | |
| 218 | |
| 219 | def print_tree(node, indent=0): |