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    CRITICAL_AMOUND = 20 
 28
 29    def __init__(self, position, initiative):
 30        super().__init__(position)
 31        self._initiative = initiative
 32        self._bag = []
 33        self._only_hosts = []
 34    
 35    def get_initiative(self):
 36        return self._initiative
 37    
 38    def get_hosts(self):
 39        return len(self._only_hosts)
 40
 41    def set_only_hosts(self, kids_and_hosts):
 42        self.__only_hosts = {host for host in kids_and_hosts if not isinstance(host, Kid)}
 43    
 44    def add_candy(self, Candy):     
 45        self._bag.append(Candy)
 46
 47    def is_critical(self):
 48        uranium_in_bag = 0
 49        for candy in self._bag:
 50            uranium_in_bag += candy.get_uranium_quantity()
 51        return uranium_in_bag > self.CRITICAL_AMOUND
 52    
 53    @staticmethod
 54    def distance(p1, p2):
 55        return math.sqrt((p1[0] - p2[0])**2 + (p1[1] - p2[1])**2)
 56
 57    def sort_all_distances(self):
 58        distance_nearest_host = []
 59        for host in self._only_hosts:
 60            distance_host_kid = self.distance(self.get_position(), host.get_position())
 61            distance_nearest_host.append((distance_host_kid, host))
 62        sorted_list = sorted(distance_nearest_host, key=lambda x: x[0])
 63        return sorted_list
 64
 65    def next_host(self):
 66        return self.sort_all_distances()[0][1]
 67        
 68    def is_hosted(self, host):                            
 69        self.set_position(host.get_position())
 70        self._only_hosts.remove(host)
 71
 72
 73class Host(Person):
 74    def __init__(self, position, candies):
 75        super().__init__(position)
 76        self._list_kids = []
 77        self._bag = [Candy(*candy) for candy in candies]
 78        self.sorted_initiative_kids = []
 79
 80    def add_kid(self, kid):
 81        self._list_kids.append(kid)
 82    def get_bag(self):
 83        return self._bag
 84    def is_empty_bag(self):
 85        return len(self._bag) < 1
 86
 87    @staticmethod
 88    def max_candy(candies):
 89        if len(candies) != 0:
 90            max_mass = candies[0].get_mass()
 91        for candy in candies:
 92            if max_mass < candy.get_mass():
 93                max_mass = candy.get_mass()
 94        for candy in candies:
 95            if candy.get_mass() == max_mass:
 96                return candy
 97
 98    def choose_kid(self):
 99        self.sorted_initiative_kids = []
100        for kid in self._list_kids:
101            self.sorted_initiative_kids.append((kid.get_initiative(), kid))
102        self.sorted_initiative_kids = sorted(self.sorted_initiative_kids, key = lambda x: x[0])
103        for _, kid in self.sorted_initiative_kids:
104            if not self.is_empty_bag():
105                candy = self.remove_candy(self.max_candy)
106                kid.add_candy(candy) 
107            kid.is_hosted(self)  
108            self._list_kids.remove(kid)
109            self.sorted_initiative_kids = self.sorted_initiative_kids[1:]                    
110
111    def remove_candy(self, max_candy):       
112        if len(self._bag) == 0:
113            return None         
114        candy_to_remove = max_candy(self._bag)               
115        self._bag.remove(candy_to_remove)
116        return candy_to_remove
117
118
119class FluxCapacitor:
120    VICTIM = 1
121
122    def __init__(self, participants):
123        self._participants = participants
124        self._victims = [] 
125        
126    def get_victim(self):
127        for participant in self._participants:
128            if isinstance(participant, Kid):
129                participant.set_only_hosts(self._participants)
130        while len(self._victims) < 1:
131            for participant in self._participants:
132                if isinstance(participant, Kid):
133                    participant.next_host().add_kid(participant)
134            for participant in self._participants:
135                if isinstance(participant, Host):
136                    participant.choose_kid()
137            for participant in self._participants:
138                if isinstance(participant, Kid) and participant.get_hosts() != 0:
139                    if participant.is_critical():
140                        self._victims.append(participant)
141            else: 
142                return None
143        return self._victims