CSES - Puun polut

Polku on kahden puun solmun välinen reitti, ja polun pituus on askelten määrä reitillä. Tehtäväsi on laskea pituuksien summa kaikista puussa olevista poluista, joissa päätesolmut ovat puun lehtiä.

Tehtäväpohjan esimerkissä polut ovat seuraavat:

Polkujen pituudet ovat 33, 33 ja 22, joten vastaus on 3+3+2=83+3+2=8.

Voit olettaa, että nn on enintään 100.

Toteuta tiedostoon treepath.py funktio count, joka palauttaa pituuksien summan.

from collections import namedtuple

def count(node):
    # TODO

if __name__ == "__main__":
    Node = namedtuple("Node",["left","right"])
    tree = Node(Node(None,None),Node(Node(None,None),Node(None,None)))
    print(count(tree)) # 8