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