lafrite/Mapytex
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Balancing a AssocialTree

Browse Source
This commit is contained in:
Bertrand Benjamin 2018-01-31 17:18:52 +03:00
parent 935a0d0d23
commit a0fa09a381
1 changed files with 96 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
mapytex/calculus/core/tree.py
View File

@ -966,6 +966,102 @@ class AssocialTree(Tree):
except AttributeError:
yield callback(self.right_value)
def balance(self):
""" Balance the tree
:example:
>>> t = AssocialTree.from_str("1+2+3+4+5+6+7+8+9")
>>> print(t)
+
> +
| > +
| | > +
| | | > +
| | | | > +
| | | | | > +
| | | | | | > +
| | | | | | | > 1
| | | | | | | > 2
| | | | | | > 3
| | | | | > 4
| | | | > 5
| | | > 6
| | > 7
| > 8
> 9
>>> bal_t = t.balance()
>>> print(bal_t)
+
> +
| > +
| | > 1
| | > 2
| > +
| | > 3
| | > 4
> +
| > +
| | > 5
| | > 6
| > +
| | > 7
| | > +
| | | > 8
| | | > 9
>>> t = AssocialTree.from_str("1*2*3*4*5+6+7+8+9")
>>> print(t)
+
> +
| > +
| | > +
| | | > *
| | | | > *
| | | | | > *
| | | | | | > *
| | | | | | | > 1
| | | | | | | > 2
| | | | | | > 3
| | | | | > 4
| | | | > 5
| | | > 6
| | > 7
| > 8
> 9
>>> bal_t = t.balance()
>>> print(bal_t)
+
> +
| > *
| | > *
| | | > 1
| | | > 2
| | > *
| | | > 3
| | | > *
| | | | > 4
| | | | > 5
| > 6
> +
| > 7
| > +
| | > 8
| | > 9
:returns: Balanced Tree (not AssocialTree)
"""
leafs = self.get_leafs()
balanced_leafs = []
for l in leafs:
try:
balanced_leafs.append(l.balance())
except AttributeError:
balanced_leafs.append(l)
t = Tree.from_list(self.node, balanced_leafs)
return t
# -----------------------------
# Reglages pour 'vim'
# vim:set autoindent expandtab tabstop=4 shiftwidth=4: