1import math
  2
  3
  4class Candy:
  5    def __init__(self, mass, uranium):
  6        self.mass = mass
  7        self.uranium = uranium
  8        
  9    def get_uranium_quantity(self):
 10        return self.mass * self.uranium
 11    
 12    def get_mass(self):
 13        return self.mass
 14    
 15    
 16class Person:
 17    def __init__(self, position):
 18        self.position = list(position)
 19        
 20    def get_position(self):
 21        return self.position
 22    
 23    def set_position(self, new_position):
 24        self.position[0] = new_position[0]
 25        self.position[1] = new_position[1]
 26        
 27        
 28class Kid(Person):
 29    def __init__(self, position, initiative):
 30        super().__init__(position)
 31        self.initiative = initiative
 32        self.bucket = []
 33        self.meeted_hosts = []
 34        
 35    def get_initiative(self):
 36        return self.initiative
 37    
 38    def add_candy(self, candy):
 39        self.bucket.append(candy)
 40        
 41    def is_critical(self) :
 42        sum = 0
 43        for candy in self.bucket :
 44            sum += candy.get_uranium_quantity()
 45        return sum > 20
 46        #return sum(candy.get_uranium_quantity() for candy in self.bucket) > 20
 47    
 48    
 49def choose_candy(bucket):
 50    return max(bucket, key=lambda candy: candy.get_mass())
 51
 52
 53class Host(Person):
 54    def __init__(self, position, candies) :
 55        super().__init__(position)
 56        self.candies = candies
 57        self.bucket = []
 58        for candy in candies:
 59            current_candy = Candy(*candy)
 60            self.bucket.append(current_candy)
 61            
 62    def remove_candy(self, choose_candy_func) :
 63        if not self.bucket:
 64            return None
 65        chosen_candy = choose_candy_func(self.bucket)
 66        self.bucket.remove(chosen_candy)
 67        return chosen_candy
 68    
 69    
 70def find_distance(kid_position, host_positon) :
 71    return math.sqrt((kid_position[0] - host_positon[0])**2 + (kid_position[1] - host_positon[1])**2)
 72
 73
 74def find_closest(kid, hosts) :
 75    if not hosts :
 76        return None
 77    closest_dist = float('inf') #това го видях от chatgpt, понеже в python няма int_max
 78    closest_host = None
 79    for host in hosts :
 80        if host not in kid.meeted_hosts and host != closest_host :
 81            curr_dist = find_distance(kid.position, host.position)
 82            if curr_dist < closest_dist :
 83                closest_dist = curr_dist
 84                closest_host = host
 85            elif curr_dist == closest_dist :
 86                curr_x, curr_y = host.get_position()
 87                closest_x, closest_y = closest_host.get_position()
 88                if curr_x < closest_x or (curr_x == closest_x and curr_y < closest_y) :
 89                    closest_dist = curr_dist
 90                    closest_host = host
 91    return closest_host    
 92    
 93    
 94class FluxCapacitor:
 95    def __init__(self, participants) :
 96        self.kids = []
 97        self.hosts = []
 98        self.victim_set = set()
 99        for human in participants :
100            if isinstance(human, Kid) :
101                self.kids.append(human)
102            elif isinstance(human, Host) :
103                self.hosts.append(human)
104                
105    def get_victim(self):
106        is_finished = False
107        if not self.kids :
108            return None
109        while not is_finished :
110            for kid in sorted(self.kids, key=lambda x: x.get_initiative(), reverse=True) :
111                closest_host = find_closest(kid, self.hosts)
112                if closest_host is None :
113                    self.kids.remove(kid)
114                    continue
115                curr_candy = closest_host.remove_candy(choose_candy)
116                if curr_candy == None :
117                    kid.meeted_hosts.append(closest_host)
118                    continue
119                kid.add_candy(curr_candy)
120                kid.meeted_hosts.append(closest_host)
121                kid.set_position(closest_host.get_position())
122                if kid.is_critical() :
123                    self.victim_set.add(kid)
124                    self.kids.remove(kid)           
125            if len(self.victim_set) != 0 : 
126                return self.victim_set
127            if not self.kids :
128                    is_finished = True
129                    break
130        return None