CSES - Esivanhemmat

Tehtäväsi on selvittää, onko solmu a solmun b esivanhempi puussa. Tämä tarkoittaa, että solmusta a pääsee solmuun b laskeutumalla alaspäin eli solmu b esiintyy solmun a alipuussa.

Esimerkiksi alla olevassa puussa solmu 1 on solmun 3 esivanhempi ja solmu 2 on solmun 6 esivanhempi.

Toteuta tiedostoon ancestors.py funktio ancestor, jolle annetaan parametrina viittaus puun juurisolmuun sekä solmujen tunnukset a ja b. Funktio palauttaa True tai False.

class Node:
    def __init__(self, value, children=None):
        self.value = value
        self.children = children if children else []

    def __repr__(self):
        if self.children == []:
            return f"Node({self.value})"
        else:
            return f"Node({self.value}, {self.children})"

def ancestor(node, a, b):
    # TODO

if __name__ == "__main__":
    tree1 = Node(1, [Node(4, [Node(3), Node(7)]),
                     Node(5),
                     Node(2, [Node(6)])])
    print(ancestor(tree1, 1, 3)) # True
    print(ancestor(tree1, 2, 6)) # True
    print(ancestor(tree1, 3, 1)) # False
    print(ancestor(tree1, 5, 6)) # False
    print(ancestor(tree1, 3, 3)) # True

    tree2 = Node(1, [Node(2, [Node(3, [Node(4)])])])
    print(ancestor(tree2, 1, 4)) # True
    print(ancestor(tree2, 3, 2)) # False

    tree3 = Node(1, [Node(2), Node(3), Node(4)])
    print(ancestor(tree3, 1, 2)) # True
    print(ancestor(tree3, 1, 1)) # True
    print(ancestor(tree3, 2, 1)) # False
    print(ancestor(tree3, 5, 5)) # False