1class Candy:
  2
  3    def __init__(self, mass, uranium):
  4        self._mass = mass
  5        self._uranium = uranium * mass
  6
  7    def get_uranium_quantity(self):
  8        return self._uranium
  9    
 10    def get_mass(self):
 11        return self._mass
 12    
 13    def __gt__(self, other):
 14        return self._mass > other._mass
 15    
 16
 17class Person:
 18    
 19    def __init__(self, position):
 20        self._position = position
 21    
 22    def get_position(self):
 23        return self._position
 24    
 25    def set_position(self, position):
 26        self._position = position
 27
 28    def get_distance_to(self, position):
 29        return (self._position[0] - position[0])**2 + (self._position[1] - position[1])**2
 30    
 31
 32class Kid(Person):
 33
 34    _CRITICAL_MASS = 20
 35
 36    def __init__(self, position, initiative):
 37        super().__init__(position)
 38        self._initiative = initiative
 39        self._candy_box = []
 40        self._current_uranium = 0
 41        self.hosts_to_visit = set()
 42    
 43    def get_initiative(self):
 44        return self._initiative
 45    
 46    def add_candy(self, candy):
 47        self._candy_box.append(candy)
 48        self._current_uranium += candy.get_uranium_quantity()
 49
 50    def is_critical(self):
 51        return self._current_uranium > self._CRITICAL_MASS
 52    
 53    def visit_closest_host(self):
 54        closest_host = min(self.hosts_to_visit, key = host_comparator(self._position))
 55        self.hosts_to_visit.remove(closest_host)
 56        self._position = closest_host.get_position()
 57        closest_host.kids.add(self)
 58    
 59    def __gt__(self, other):
 60        return self._initiative > other._initiative 
 61    
 62    
 63
 64class Host(Person):
 65
 66    def __init__(self, position, candy_info):
 67        super().__init__(position)
 68        self._candy_box = set(map(lambda info: Candy(*info), candy_info))
 69        self.kids = set()
 70
 71    def remove_candy(self, func):
 72        if not self._candy_box:
 73            return None
 74        result = func(self._candy_box)
 75        self._candy_box.remove(result)
 76        return result
 77    
 78    def give_candy(self):
 79        while self.kids:
 80            kid = max(self.kids)
 81            candy = self.remove_candy(max)
 82            kid.add_candy(candy)
 83            self.kids.remove(kid)
 84
 85
 86class FluxCapacitor:
 87
 88    def __init__(self, persons):
 89        self._kids = set(filter(lambda p: isinstance(p, Kid), persons))
 90        self._hosts = set(filter(lambda p: isinstance(p, Host), persons))
 91    
 92    def get_victim(self):
 93        for kid in self._kids:
 94            kid.hosts_to_visit = set(self._hosts)
 95        
 96        victims = set()
 97        for _ in self._hosts:
 98            for kid in self._kids:
 99                kid.visit_closest_host()
100            for host in self._hosts:
101                host.give_candy()
102            for kid in self._kids:
103                if kid.is_critical():
104                    victims.add(kid)
105            if victims:
106                return victims
107        return None
108    
109           
110def host_comparator(kid_position):
111    def compare(host):
112        return (host.get_distance_to(kid_position), host.get_position()[0])
113    return compare