1import math
  2
  3class Candy:
  4    def __init__(self, mass, uranium):
  5        self.mass = mass
  6        self.uranium = uranium
  7    
  8    def get_uranium_quantity(self):
  9        return self.mass * self.uranium
 10    
 11    def get_mass(self):
 12        return self.mass
 13
 14
 15class Person:
 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, position):
 23        self.position = position
 24        
 25
 26class Kid(Person):
 27
 28    MAX_URANIUM_QUANTITY = 20
 29
 30    def __init__(self, position, initiative):
 31        super().__init__(position)
 32        self.initiative = initiative
 33        self.visited_hosts = []
 34        self.basket = []
 35    
 36    def get_initiative(self):
 37        return self.initiative
 38    
 39    def add_candy(self, candy):
 40        self.basket.append(candy)
 41
 42    def if_host_visited(self, host):
 43        found = False
 44        for visited_host in self.visited_hosts:
 45            if visited_host == host:
 46                found = True
 47                break
 48
 49        return found
 50    
 51    def is_critical(self):
 52        uranium_quantity = 0
 53
 54        for candy in self.basket:
 55            uranium_quantity = uranium_quantity + candy.get_uranium_quantity()
 56        
 57        return uranium_quantity > self.MAX_URANIUM_QUANTITY
 58
 59
 60class Host(Person):
 61    
 62    def __init__(self, position, candies = []):
 63        super().__init__(position)
 64        self.candies = []
 65        for tup in candies:
 66            self.candies.append(Candy(tup[0], tup[1]))
 67    
 68    def remove_candy(self, func):
 69
 70        if not self.candies:
 71            return None
 72        
 73        candy_to_remove = func(self.candies)
 74        self.candies.remove(candy_to_remove)
 75
 76        return candy_to_remove
 77
 78    def add_host(self, host):
 79        self.visited_hosts.append(host)
 80    
 81
 82
 83    
 84
 85class FluxCapacitor:
 86    
 87    def __init__(self, participants):
 88        self.participants = participants
 89
 90    def __remove_biggest_mass_candy(self, candies):
 91        return max(candies, key=lambda candy: candy.get_mass())
 92
 93    def __get_closest_host(self, hosts, kid):
 94        closest_host = None
 95        min_distance = float('inf')
 96
 97        for host in hosts:
 98
 99            if kid.if_host_visited(host):
100                continue
101
102            xi, yi = host.get_position()
103            x1, y1 = kid.get_position()
104            distance = math.sqrt((x1 - xi)**2 + (y1 - yi)**2)
105            if distance < min_distance:
106                min_distance = distance
107                closest_host = host
108            elif min_distance == distance and closest_host is not None:
109                if host.get_position[0] < closest_host.get_position[0]:
110                    closest_host = host
111                elif host.get_position[0] == closest_host.get_position[0] and host.get_position[1] < closest_host.get_position[1]:
112                    closest_host = host
113        
114        return closest_host
115    
116    def get_kids_critical(self, kids):
117
118        result = []
119
120        for kid in kids:
121            if kid.is_critical():
122                result.append(kid)
123        
124        return result
125    
126    def get_victim(self):
127        
128        kids = set()
129        hosts = set()
130
131        for participant in self.participants:
132            if isinstance(participant, Kid):
133                kids.add(participant)
134            else:
135                hosts.add(participant)
136
137        counter_to_finish = 0
138        while True:
139            counter_to_finish = counter_to_finish + 1
140            for kid in kids:
141                host = self.__get_closest_host(hosts, kid)
142
143                if host is not None:
144                    candy = host.remove_candy(self.__remove_biggest_mass_candy)
145                    if candy is not None:
146                        kid.add_candy(candy)
147
148            result = self.get_kids_critical(kids)
149            if result:
150                return result
151            else:
152                counter = 0
153                for kid in kids:
154                    if len(kid.visited_hosts) == len(hosts):
155                        counter = counter + 1
156                
157                if counter == len(kids) or counter_to_finish == len(hosts):
158                    return None
159
160            
161
162candy = Candy(20, 1.2)
163person = Person((1, 2))
164kid = Kid((0, 0), 123)
165host = Host((3, 4), [(100, 1.0), (200, 0.5)])
166flux_capacitor = FluxCapacitor({kid, host})
167flux_capacitor.get_victim()