1class Candy:
  2    
  3    def __init__(self, mass, uranium):
  4        self.mass = mass
  5        self.uranium = uranium
  6    
  7    def get_uranium_quantity(self):
  8        return self.mass * self.uranium
  9    
 10    def get_mass(self):
 11        return self.mass
 12
 13
 14class Person:
 15    
 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, new_position):
 23        self.position = new_position
 24
 25
 26class Kid(Person):
 27    
 28    CRITICAL_MASS = 20
 29    
 30    def __init__(self, position, initiative):
 31        super().__init__(position)
 32        self.initiative = initiative
 33        self.not_visited = set()
 34        self.basket = set()
 35        
 36    def get_initiative(self):
 37        return self.initiative
 38
 39    def add_candy(self, candy):
 40        self.basket.add(candy)
 41    
 42    def is_critical(self):
 43        uranium_in_basket = 0
 44        for candy in self.basket:
 45            uranium_in_basket += candy.get_uranium_quantity()
 46        return uranium_in_basket > self.CRITICAL_MASS
 47
 48    def set_not_visited_hosts(self, hosts):
 49        self.not_visited = {host for host in hosts}
 50    
 51    def closest_host(self, hosts): #остава си така :/
 52        distances_by_X = {host : abs(self.position[0] - host.position[0]) for host in hosts}
 53        closest_by_X = {key : value for key, value in distances_by_X.items() if value == min(distances_by_X.values())}
 54        distances = {host : abs(self.position[1] - host.position[1]) for host in closest_by_X}
 55        closest_host = [key for key, value in distances.items() if value == min(distances.values())]
 56        return closest_host[0]
 57                
 58
 59class Host(Person):
 60    
 61    def __init__(self, position, candies):
 62        super().__init__(position)
 63        self.candies = candies
 64        self.visited_by = set()
 65    
 66    def remove_candy(self, func):
 67        if len(self.candies) == 0:
 68            return
 69        return func
 70    
 71    def set_visited_by(self, kids):
 72        self.visited_by = {kid for kid in kids if self.position == kid.position}
 73
 74
 75class FluxCapacitor:
 76    
 77    def __init__(self, participants):
 78        self.participants = participants
 79        self.hosts = {person for person in self.participants if type(person) is Host}
 80        self.kids = {person for person in self.participants if type(person) is Kid}
 81    
 82    @staticmethod
 83    def which_candy(candies):
 84        this_candy = max(candies, key=lambda candy: candy.mass)
 85        candies.remove(this_candy)
 86        return this_candy
 87    
 88    def get_victim(self):
 89        victims = set()
 90                
 91        for kid in self.kids: #създавам списък с непосетени домакини за всяко дете
 92            kid.set_not_visited_hosts(self.hosts)
 93            
 94        for _ in range(len(self.hosts)):
 95            for kid in self.kids: #всяко дете се мести до най-близкия си домакин и го "отметва" като посетен
 96                closest_host = kid.closest_host(kid.not_visited)
 97                kid.set_position(closest_host.position)
 98                kid.not_visited.remove(closest_host)
 99                
100            for host in self.hosts: #раздаване на бонбони
101                if len(host.candies) == 0: #ако домакинът няма бонбони, го "подминаваме"
102                    continue
103                
104                remove = host.remove_candy(self.which_candy)
105                host.set_visited_by(self.kids) #сортирам децата, посетили даден домакин, низходящо по инициатива
106                sorted_initiative = sorted(host.visited_by, key=lambda kid : kid.initiative, reverse=True)
107                
108                for kid in sorted_initiative:
109                    kid.add_candy(remove(host.candies)) #даваме бонбон
110                    if kid.is_critical(): #проверявам дали детето е радиоактивно
111                        victims.add(kid)
112                         
113            if len(victims) != 0: #първата инстанция, в която victims не е празен, спираме играта
114                return victims
115   
116        return