1CRITICAL_MASS = 20
  2
  3class Candy:
  4    def __init__(self, mass, uranium):
  5        self.mass = mass
  6        self.uranium = uranium
  7
  8    def get_uranuium_quantity(self):
  9        return self.uranium * self.mass
 10    
 11    def get_mass(self):
 12        return self.mass
 13    
 14
 15class Person:
 16    def __init__(self, position):
 17        self.position = position
 18
 19    def get_position(self):
 20        return self.position
 21    
 22    def set_position(self, position):
 23        self.position = position
 24
 25
 26class Kid(Person):
 27    def __init__(self, position, initiative):
 28        super().__init__(position)
 29        self.initiative = initiative
 30        self.basket = []
 31
 32    def get_initiative(self):
 33        return self.initiative
 34    
 35    def add_candy(self, new_candy):
 36        self.basket.append(new_candy)
 37
 38    def is_critical(self):  
 39        return sum(self.basket) > CRITICAL_MASS
 40    
 41
 42class Host(Person):
 43    def __init__(self, position, candies):
 44        super().__init__(position)
 45        self.candies = candies
 46
 47    def remove_candy(self, func):
 48        if len(self.candies) == 0:
 49            return None
 50        
 51        return func(self.candies)
 52    
 53    def get_candies_count(self, candies):
 54        return len(candies)
 55    
 56    def greater_distance(self, other_person):
 57        return ((self.get_position()[0] - other_person.get_position()[0]) ** 2 + (self.get_position()[1] - other_person.get_position()[1]) ** 2)
 58    
 59
 60class FluxCapacitor:
 61    def __init__(self, participants):
 62        self.participants = participants
 63        self.kids = []
 64        self.hosts = []
 65        for participant in self.participants:
 66            if participant is Kid:
 67                self.kids.append(participant)
 68            elif participant is Host:
 69                self.hosts.append(participant)
 70
 71        self.kids.sort(key=lambda x: x.get_initiative(), reverse = True)
 72
 73        every_kid_hosts = {}
 74        for kid in self.kids:
 75            every_kid_hosts[kid] = self.hosts
 76
 77    def remove_function(self, current_candies): 
 78        current_candies.sort(key=lambda x: x.mass, reverse = True)
 79        best_candy = current_candies.pop(0)
 80        return best_candy 
 81
 82    def get_min_host(self, kid, hosts):
 83        return min(hosts, key=(lambda host:(host.greater_distance(kid), host.get_position()[0], host.get_position()[1])))
 84
 85    def get_victim(self):
 86        victims = []
 87
 88        while len(self.hosts) != 0:
 89            for kid in self.kids:
 90                best_host = self.get_min_host(kid, self.every_kid_hosts[kid])
 91
 92                best_candy = best_host.remove_candy(self.remove_function(best_host.candies))
 93                kid.add_candy(best_candy)
 94
 95                if kid.is_critical():
 96                    victims.append(kid)
 97
 98                kid.set_position(best_host.get_position())
 99
100                for host in self.hosts:
101                    if host.get_candies_count() == 0:
102                        self.hosts.discard(host)
103
104                self.every_kid_hosts[kid].discard(best_host)
105
106            if len(victims) != 0:
107                return victims
108
109        return None