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 = position
 19        self.basket_of_sweets = []
 20
 21    def  get_position(self):
 22        return self.position  
 23
 24    def set_position(self, new_position):
 25        self.position = new_position
 26        return self.position
 27
 28
 29class Kid(Person):
 30    def __init__(self, position, initiative):
 31        super().__init__(position)
 32        self.initiative = initiative
 33        self.history_of_positions = []
 34    
 35    def get_initiative(self):
 36        return self.initiative
 37    
 38    def add_candy(self, sweet):
 39        self.basket_of_sweets.append(sweet)
 40
 41    def is_critical(self):
 42        uranium_q_ty = 0
 43        critical_uranium_q_ty = 20
 44
 45        for sweet in self.basket_of_sweets:
 46            uranium_q_ty += sweet.get_uranium_quantity()
 47        return uranium_q_ty > critical_uranium_q_ty
 48
 49
 50class Host(Person):
 51    def __init__(self, position, candies):
 52        super().__init__(position)
 53        self.candies = candies
 54        
 55        for candy in self.candies:
 56            self.basket_of_sweets.append(Candy(*candy))
 57    
 58    def remove_candy(self, func):
 59        if not self.basket_of_sweets:
 60            return None
 61        candy = func(self.basket_of_sweets)
 62
 63        self.basket_of_sweets.remove(candy)
 64        return candy
 65    
 66
 67class FluxCapacitor:
 68    def __init__(self, participants):
 69        self.participants = participants
 70        self.kids = []
 71        self.hosts = []
 72
 73        for person in participants:
 74            if type(person) is Kid:
 75                self.kids.append(person)
 76            elif type(person) is Host:
 77                self.hosts.append(person)
 78
 79    def set_new_position(self):
 80        for kid in self.kids:
 81            distance = 0
 82            new_position = ()
 83            list_of_distances_and_positions = []
 84            for host in self.hosts:
 85                if host.get_position() in kid.history_of_positions:
 86                    continue
 87                distance = math.dist(kid.get_position(), host.get_position())
 88                list_of_distances_and_positions.append((distance, host.get_position()))
 89  
 90            new_position = min(list_of_distances_and_positions)[1]
 91            kid.set_position(new_position) 
 92            kid.history_of_positions.append(kid.get_position())
 93
 94    @staticmethod
 95    def biggest_sweet(basket_of_sweets):
 96        biggest_sweet = max(basket_of_sweets, key=lambda x: x.mass)
 97        return biggest_sweet
 98
 99    def recieve_candy(self):
100        for host in self.hosts:
101            kids_with_this_host = []
102
103            for kid in self.kids:
104                if kid.get_position() == host.get_position():
105                    kids_with_this_host.append(kid)
106            
107            kids_with_this_host.sort(key=lambda x: x.initiative, reverse = True)
108            
109            for kid in kids_with_this_host:
110                sweet = host.remove_candy(self.biggest_sweet)
111                if sweet:
112                    kid.add_candy(sweet)
113
114    def take_kids_to_heaven(self):
115        kids_in_heaven = set()
116
117        for kid in self.kids:
118            if kid.is_critical():
119                kids_in_heaven.add(kid)
120
121        return kids_in_heaven
122        
123    def get_victim(self):
124        for host in self.hosts:
125            self.set_new_position()
126            self.recieve_candy()
127            kids_in_heaven = self.take_kids_to_heaven()
128           
129            if kids_in_heaven:
130                return kids_in_heaven
131            
132        return None