Tehtäväsi on laskea, montako solmua on annetussa binääripuussa tietyllä tasolla. Puun juuri on tasolla , sen lapset ovat tasolla , jne. Voit olettaa, että puussa on enintään 100 solmua.
Tehtäväpohjassa on esimerkkinä seuraava puu:
Toteuta tiedostoon samelevel.py
funktio count
, joka palauttaa solmujen määrän.
from collections import namedtuple def count(node, level): # TODO if __name__ == "__main__": Node = namedtuple("Node", ["left", "right"]) tree = Node(None,Node(Node(None,None),Node(None,None))) print(count(tree,1)) # 1 print(count(tree,2)) # 1 print(count(tree,3)) # 2 print(count(tree,4)) # 0