1import math
  2
  3
  4class Candy:
  5
  6    def __init__(self, mass: int, uranium: int):
  7        self.mass = mass
  8        self.uranium = uranium
  9
 10    def get_uranium_quantity(self):
 11        return (self.mass * self.uranium)
 12
 13    def get_mass(self):
 14        return self.mass
 15
 16
 17class Person:
 18
 19    def __init__(self, position: tuple):
 20        self.position = position
 21
 22    def get_position(self):
 23        return self.position
 24
 25    def set_position(self, new_position: tuple):
 26        self.position = new_position
 27
 28
 29class Kid(Person):
 30    CRITICAL_MASS = 20
 31
 32    def __init__(self, position: tuple, initiative: int):
 33        self.position = position
 34        self.initiative = initiative
 35        self.candies = []
 36        self.visited = set()
 37
 38    def get_initiative(self):
 39        return self.initiative
 40
 41    def add_candy(self, candy: Candy):
 42        self.candies.append(candy)
 43
 44    def is_critical(self):
 45        uranium_mass = 0
 46        for candy in self.candies:
 47            uranium_mass += candy.get_uranium_quantity()
 48
 49        return uranium_mass > self.CRITICAL_MASS
 50
 51
 52class Host(Person):
 53
 54    def __init__(self, position: tuple, candies: list):
 55        self.candies = []
 56        for candy in candies:
 57            self.candies.append(Candy(candy[0], candy[1]))
 58        self.position = position
 59
 60    def remove_candy(self, condition):
 61        if len(self.candies) == 0:
 62            return None
 63        candy = condition(self.candies)
 64        self.candies.remove(candy)
 65        return candy
 66
 67
 68class FluxCapacitor:
 69
 70    def __init__(self, participants: set):
 71        self.kids = []
 72        self.hosts = []
 73
 74        for person in participants:
 75            if type(person) is Kid:
 76                self.kids.append(person)
 77            else:
 78                self.hosts.append(person)
 79        self.kids.sort(reverse=True, key=lambda kid: kid.get_initiative())
 80
 81    def get_victim(self):
 82        dead_kids = set()
 83        if len(self.kids) == 0:
 84            return None
 85
 86        while True:
 87            for kid in self.kids:
 88                closest_host = None
 89                distance = float('inf')
 90                for host in self.hosts:
 91                    if host in kid.visited:
 92                        continue
 93                    new_distance = math.dist(kid.get_position(), host.get_position())
 94                    if distance > new_distance:
 95                        distance = new_distance
 96                        closest_host = host
 97                    elif distance == new_distance:
 98                        if closest_host.get_position() > host.get_position():
 99                            closest_host = host
100
101                if closest_host is None:
102                    return None
103                taken_candy = closest_host.remove_candy(determine_max_mass)
104
105                if taken_candy is not None:
106                    kid.add_candy(taken_candy)
107                kid.visited.add(closest_host)
108
109                if kid.is_critical():
110                    dead_kids.add(kid)
111
112            if dead_kids:
113                return dead_kids
114
115
116def determine_max_mass(candies: list):
117    max = 0
118    for candy in candies:
119        if max < candy.get_mass():
120            max = candy.get_mass()
121            max_candy = candy
122
123    return max_candy