Samat painot

Tehtäväsi on tutkia, onko annetun suuntaamattoman verkon jokaisella virittävällä puulla sama paino.

Toteuta tiedostoon sameweight.py luokka SameWeight, jossa on seuraavat metodit:

• add_edge lisää solmujen $a$ ja $b$ välille kaaren, jonka paino on $x$
• check ilmoittaa, onko verkon kaikilla virittävillä puilla sama paino

Jos verkko ei ole yhtenäinen, metodin check tulee palauttaa tulos True, koska tässä tapauksessa ei ole mitään virittävää puuta.

class SameWeight:
    def __init__(self, n):
        # TODO

    def add_edge(self, a, b, x):
        # TODO

    def check(self):
        # TODO

if __name__ == "__main__":
    same_weight = SameWeight(4)

    same_weight.add_edge(1, 2, 2)
    same_weight.add_edge(1, 3, 3)
    print(same_weight.check()) # True

    same_weight.add_edge(1, 4, 3)
    print(same_weight.check()) # True

    same_weight.add_edge(3, 4, 3)
    print(same_weight.check()) # True

    same_weight.add_edge(2, 4, 1)
    print(same_weight.check()) # False