1class Candy:
  2    def __init__(self, mass, uranium):
  3        self.mass = mass
  4        self.uranium = uranium
  5        
  6    def get_uranium_quantity(self):
  7        return self.mass * self.uranium
  8    
  9    def get_mass(self):
 10        return self.mass
 11    
 12
 13class Person:
 14    def __init__(self, position):
 15        self.position = position
 16        
 17    def get_position(self):
 18        return self.position
 19    
 20    def set_postition(self, new_position):
 21        self.position = new_position
 22        
 23    def get_distance(self, other_person):
 24        return ((self.position[0] - other_person.position[0]) ** 2 + (self.position[1] - other_person.position[1]) ** 2) ** (1/2)
 25        
 26
 27class Kid(Person):
 28    MAX_URANIUM = 20
 29    def __init__(self, position, initiative):
 30        self.list_of_candies = []
 31        self.list_of_hosts = []
 32        self.sum_of_candies = 0
 33        super().__init__(position)
 34        self.initiative = initiative
 35    
 36    def get_initiative(self):
 37        return self.initiative
 38    
 39    def add_candy(self, candy):
 40        if candy is not None:
 41            self.candy = candy
 42            self.list_of_candies.append(candy)
 43        
 44    def go_to_host(self):
 45        if not self.list_of_hosts:
 46            return None
 47        
 48        min_dist = self.get_distance(self.list_of_hosts[0])
 49        min_host = self.list_of_hosts[0]
 50        
 51        for host in self.list_of_hosts:
 52            next_dist = self.get_distance(host)
 53            if min_dist > next_dist:
 54                min_dist = next_dist
 55                min_host = host
 56            elif min_dist == next_dist:
 57                if host.position[0] < min_host.position[0]:
 58                    min_dist = next_dist
 59                    min_host = host
 60                elif host.position[0] == min_host.position[0]:
 61                    if host.position[1] < min_host.position[1]:
 62                        min_dist = next_dist
 63                        min_host = host
 64        
 65        self.add_candy(min_host.remove_candy(Host.function_for_removal))
 66        self.position = min_host.position
 67        self.list_of_hosts.remove(min_host)
 68        
 69    def is_critical(self):
 70        self.sum_of_candies = 0
 71        for candy in self.list_of_candies:
 72            self.sum_of_candies += candy.get_uranium_quantity()
 73        return self.sum_of_candies > self.MAX_URANIUM
 74    
 75
 76class Host(Person):
 77    
 78    @staticmethod
 79    def function_for_removal(my_list_of_candies):
 80        return max(my_list_of_candies, key=lambda candy: candy.get_mass())
 81
 82    def __init__(self, position, candies):
 83        super().__init__(position)
 84        self.candies = []
 85        
 86        for candy in candies:
 87            self.candies.append(Candy(*candy))
 88        
 89    def remove_candy(self, remove_function):
 90        if len(self.candies) == 0:
 91            return None
 92        else:
 93            candy_to_remove = remove_function(self.candies)
 94            self.candies.remove(candy_to_remove)
 95            return candy_to_remove
 96      
 97      
 98class FluxCapacitor:
 99    def __init__(self, set_of_people):
100        self.set_of_dead = set()
101        self.list_of_children = []
102        self.list_of_hosts = []
103        
104        for person in set_of_people:
105            if isinstance(person, Kid):
106                self.list_of_children.append(person)
107            else:
108                self.list_of_hosts.append(person)
109                
110        for kid in self.list_of_children:
111            for host in self.list_of_hosts:
112                kid.list_of_hosts.append(host)
113        
114        self.list_of_children.sort(reverse=True, key=lambda kid: kid.initiative)
115    
116    def get_victim(self):
117        while True:
118            is_there_someone_to_visit = False
119            
120            for kid in self.list_of_children:
121                if len(kid.list_of_hosts) != 0:
122                    is_there_someone_to_visit = True
123                    break
124                
125            if is_there_someone_to_visit == False:
126                return None
127            
128            for kid in self.list_of_children:
129                kid.go_to_host()
130            
131            for kid in self.list_of_children:
132                if kid.is_critical():
133                    self.set_of_dead.add(kid)
134                    
135            if len(self.set_of_dead) != 0:
136                return self.set_of_dead
137