CSES - Tanssiaiset

Tanssiaisiin osallistuu nn opiskelijaa Kumpulasta ja nn opiskelijaa Viikistä. Molempien kampusten opiskelijat on numeroitu 1,2,,n1,2,\dots,n.

Jokaisessa tanssiparissa toinen opiskelija on Kumpulasta ja toinen Viikistä, ja tiedät kaikki parit, jotka suostuvat tanssimaan keskenään. Jokainen opiskelija voi kuulua enintään yhteen tanssipariin. Mikä on suurin määrä pareja, jotka on mahdollista muodostaa?

Voit olettaa, että nn on enintään 5050 ja luokan metodeita kutsutaan enintään 100100 kertaa.

Toteuta tiedostoon ball.py luokka Ball, jossa on seuraavat metodit:

  • konstruktori, jolle annetaan määrä nn
  • add_pair määrittää, että Kumpulan opiskelija aa ja Viikin opiskelija bb suostuvat tanssimaan keskenään
  • calculate ilmoittaa suurimman mahdollisen tanssiparien määrän
class Ball:
    def __init__(self,n):
        # TODO

    def add_pair(self,a,b):
        # TODO

    def calculate(self):
        # TODO

if __name__ == "__main__":
    b = Ball(4)
    print(b.calculate()) # 0
    b.add_pair(1,2)
    print(b.calculate()) # 1
    b.add_pair(1,3)
    b.add_pair(3,2)
    print(b.calculate()) # 2