CSES - Sama syvyys

Tehtäväsi on tarkastaa, onko jokaisen puussa olevan lehden syvyys sama.

Toteuta tiedostoon samedepth.py funktio check, joka tarkastaa asian.

from collections import namedtuple

def check(node):
    # TODO

if __name__ == "__main__":
    Node = namedtuple("Node", ["children"], defaults=[[]])

    tree1 = Node([
                Node(),
                Node([Node([Node(), Node()])]),
                Node([Node(), Node()])
            ])

    tree2 = Node([Node([Node()]), Node([Node(), Node()])])

    print(check(tree1)) # False
    print(check(tree2)) # True