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        setattr(self, self.position, new_position)
 24        
 25    def get_dist(self, other_person):
 26        return ((self.position[0] - other_person.position[0])**2 + ((self.position[1] - other_person.position[1])**2))**(1/2)
 27
 28        
 29class Kid(Person):
 30    
 31    def __init__(self, position, initiative):
 32        super().__init__(position)
 33        self.initiative = initiative
 34        self.basket = []
 35        self.hosts_to_visit = []
 36        self.uranium_in_basket = 0
 37        
 38    @staticmethod
 39    def get_candy_with_most_mass(candys):
 40        return max(candys, key=lambda c: c.mass)
 41
 42    def get_initiative(self):
 43        return self.initiative
 44    
 45    def add_candy(self, candy):
 46        if candy is not None:
 47            self.basket.append(candy)
 48            self.uranium_in_basket += candy.get_uranium_quantity()
 49        
 50    def is_critical(self):
 51        return self.uranium_in_basket > 20
 52    
 53    def fill_hosts_to_visit(self, hosts):
 54        for host in hosts:
 55            self.hosts_to_visit.append(host)
 56        
 57    def go_to_closest_host(self):
 58        if not self.hosts_to_visit:
 59            return
 60        
 61        min_dist = None
 62        for host in self.hosts_to_visit:
 63            if not min_dist:
 64                min_dist = self.get_dist(self.hosts_to_visit[0])
 65                min_dist_host = self.hosts_to_visit[0]
 66            if (curr_dist := self.get_dist(host)) < min_dist:
 67                min_dist = curr_dist
 68                min_dist_host = host
 69            elif (curr_dist := self.get_dist(host)) == min_dist:
 70                if host.position[0] < min_dist_host.position[0]:
 71                    min_dist_host = host
 72                elif host.position[0] == min_dist_host.position[0]:
 73                    if host.position[1] < min_dist_host.position[1]:
 74                        min_dist_host = host
 75        
 76        self.add_candy(min_dist_host.remove_candy(Kid.get_candy_with_most_mass))
 77        self.position = host.get_position()
 78        self.hosts_to_visit.remove(host)
 79        
 80
 81class Host(Person):
 82    
 83    def __init__(self, position, candies):
 84        super().__init__(position)
 85        self.candies = [Candy(*candy) for candy in candies]
 86        
 87    def remove_candy(self, func):
 88        if len(self.candies) == 0:
 89            return None
 90        
 91        candy = func(self.candies)
 92        self.candies.remove(candy)
 93        return candy
 94    
 95
 96class FluxCapacitor:
 97    
 98    def __init__(self, participants):
 99        self.kids = list(filter(lambda x: type(x) is Kid, participants))
100        self.hosts = list(filter(lambda x: type(x) is Host, participants))
101        self.kids.sort(reverse=True, key=lambda kid: kid.initiative)
102        
103    def get_victim(self):
104        
105        for kid in self.kids:
106            kid.fill_hosts_to_visit(self.hosts)
107        
108        dead_children = set()
109        while True:
110            no_more_hosts_to_visit = True
111            for kid in self.kids:
112                if kid.hosts_to_visit:
113                    no_more_hosts_to_visit = False
114                    break
115            
116            if no_more_hosts_to_visit:
117                return None
118            
119            for kid in self.kids:
120                kid.go_to_closest_host()
121            
122            for kid in self.kids:
123                if kid.is_critical():
124                    dead_children.add(kid)
125            
126            if len(dead_children) > 0:
127                return dead_children