1import math
  2import sys
  3
  4
  5class Candy:
  6    def __init__(self, mass, uranium):
  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    def __init__(self, position):
 19        self.x, self.y = position
 20    
 21    def get_position(self):
 22        return self.x, self.y
 23    
 24    def set_position(self, new_position):
 25        self.x = new_position[0]
 26        self.y = new_position[1]
 27
 28
 29class Kid(Person):
 30    
 31    MAX_URANIUM = 20
 32    
 33    def __init__(self, position, initiative):
 34        super().__init__(position)
 35        self.initiative = initiative
 36        self.mass_counter = 0
 37        
 38    def get_initiative(self):
 39        return self.initiative
 40    
 41    def add_candy(self, candy):
 42        self.mass_counter += candy.get_uranium_quantity()
 43                
 44    def is_critical(self):
 45        return self.mass_counter > self.MAX_URANIUM
 46    
 47
 48def get_distance(pos1, pos2):
 49    return math.sqrt((pos2[0] - pos1[0])**2 + (pos2[1] - pos1[1])**2)
 50
 51
 52class Host(Person):
 53    def __init__(self, position, candies):
 54        super().__init__(position)
 55        self.basket = []
 56        
 57        for element in candies:
 58            self.basket.append(Candy(*element))
 59
 60    def remove_candy(self, func):
 61        if self.basket.count == 0:
 62            return None
 63        else:
 64            result = func(self.basket)
 65            self.basket.remove(result)
 66            return result
 67        
 68
 69class FluxCapacitor:
 70    def __init__(self, participants):
 71        self.kids_arr = [element for element in participants if isinstance(element, Kid)]
 72        self.kids_arr.sort(key=lambda x: x.get_initiative(), reverse=True)
 73        self.hosts_arr = [element for element in participants if isinstance(element, Host)]
 74        
 75    def get_victim(self):
 76        def decide_which_candy_to_give(container):
 77            result = container[0]
 78            for element in container:
 79                if element.get_mass() > result.get_mass():
 80                    result = element
 81            return result
 82
 83        host_list = [[]for _ in range(len(self.hosts_arr))]
 84        passed_hosts = [[]for _ in range(len(self.kids_arr))]
 85        flag = True
 86        while flag: 
 87            for kid in self.kids_arr:
 88                cur_distance = sys.maxsize
 89                best_host_idx = 0
 90                
 91                for i in range(len(self.hosts_arr)):
 92                    if i in passed_hosts[self.kids_arr.index(kid)]:
 93                        continue
 94                    if get_distance(kid.get_position(), self.hosts_arr[i].get_position()) < cur_distance:
 95                        cur_distance = get_distance(kid.get_position(), self.hosts_arr[i].get_position())
 96                        best_host_idx = i
 97                    elif get_distance(kid.get_position(), self.hosts_arr[i].get_position()) == cur_distance:
 98                        if self.hosts_arr[i].get_position()[0] < self.hosts_arr[best_host_idx].get_position()[0]:
 99                            best_host_idx = i
100                            cur_distance = get_distance(kid.get_position(), self.hosts_arr[best_host_idx].get_position())
101                        elif self.hosts_arr[i].get_position()[0] == self.hosts_arr[best_host_idx].get_position()[0]:
102                            if self.hosts_arr[i].get_position()[1] < self.hosts_arr[best_host_idx].get_position()[1]:
103                                best_host_idx = i
104                                cur_distance = get_distance(kid.get_position(), self.hosts_arr[best_host_idx].get_position())
105                host_list[best_host_idx].append(kid)
106                passed_hosts[self.kids_arr.index(kid)].append(best_host_idx)
107                
108            for i, host in enumerate(host_list):
109                for kid in host:
110                    kid.add_candy(self.hosts_arr[i].remove_candy(decide_which_candy_to_give))
111                    kid.set_position(self.hosts_arr[i].get_position())
112                    
113            dead_kids = set(kid for kid in self.kids_arr if kid.is_critical())
114            if dead_kids:
115                return dead_kids
116            
117            counter = 0
118            for element in passed_hosts:
119                counter += len(element)
120            
121            if counter // len(self.hosts_arr) == len(self.kids_arr):
122                return None