Sama taso

Tehtäväsi on laskea, montako solmua on annetussa binääripuussa tietyllä tasolla. Puun juuri on tasolla 1, sen lapset ovat tasolla 2, jne. Voit olettaa, että puussa on enintään 100 solmua.

Tehtäväpohjassa on esimerkkinä seuraava puu:

Tässä tasolla 1 on yksi solmu, tasolla 2 on yksi solmu, tasolla 3 on kaksi solmua ja muilla tasoilla ei ole mitään.

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