1class Candy:
  2    def __init__(self, mass, uranium):
  3        if not isinstance(mass, (int, float)) or mass <= 0:
  4            raise ValueError("Mass must be a positive number.")
  5        if not isinstance(uranium, (int, float)) or uranium < 0 or uranium > 1:
  6            raise ValueError("Uranium must be a number between 0 and 1.")
  7        
  8        self.mass = mass
  9        self.uranium = uranium
 10
 11    def get_uranium_quantity(self):
 12        return self.mass * self.uranium
 13
 14    def get_mass(self):
 15        return self.mass
 16    
 17class Person:
 18    def __init__(self, position):
 19        if not isinstance(position, tuple) or len(position) != 2 or not all(isinstance(coord, int) for coord in position):
 20            raise ValueError("Position must be a tuple of two integers.")
 21        self.position = position
 22
 23    def get_position(self):
 24        return self.position
 25
 26    def set_position(self, new_position):
 27        if not isinstance(new_position, tuple) or len(new_position) != 2 or not all(isinstance(coord, int) for coord in new_position):
 28            raise ValueError("The new position must be a tuple of two integers.")
 29        self.position = new_position
 30
 31class Kid(Person):
 32    CRITICAL_URANIUM_THRESHOLD = 20  
 33
 34    def __init__(self, position, initiative):
 35        super().__init__(position)
 36        if not isinstance(initiative, int) or initiative < 0:
 37            raise ValueError("Initiative must be a non-negative integer.")
 38        self.initiative = initiative
 39        self.candy_basket = []
 40
 41    def get_initiative(self):
 42        return self.initiative
 43
 44    def add_candy(self, candy):
 45        if not isinstance(candy, Candy):
 46            raise ValueError("You must add an object of class Candy to your basket.")
 47        self.candy_basket.append(candy)
 48
 49    def is_critical(self):
 50        uranium_quantity = sum(candy.get_uranium_quantity() for candy in self.candy_basket)
 51        return uranium_quantity > self.CRITICAL_URANIUM_THRESHOLD
 52    
 53class Host(Person):
 54    def __init__(self, position, candies):
 55        super().__init__(position)
 56        self.candy_basket = [Candy(mass, uranium) for mass, uranium in candies]
 57
 58    def remove_candy(self, select_candy_function):
 59        if not self.candy_basket:
 60            return None
 61        selected_candy = select_candy_function(self.candy_basket)
 62        self.candy_basket.remove(selected_candy)
 63        return selected_candy
 64
 65class FluxCapacitor:
 66    def __init__(self, participants):
 67        self.participants = participants
 68
 69    def get_victim(self):
 70        while True:
 71            # Групиране на децата и домакините
 72            kids = {participant for participant in self.participants if isinstance(participant, Kid)}
 73            hosts = {participant for participant in self.participants if isinstance(participant, Host)}
 74            
 75            # Да се провери дали има достатъчно уран в кошниците на децата
 76            critical_kids = {kid for kid in kids if kid.is_critical()}
 77            
 78            # Ако има критични деца, те стават жертви и играта приключва
 79            if critical_kids:
 80                return critical_kids
 81
 82            # Ако няма критични деца, процесът продължава:
 83            new_participants = set()
 84
 85            for host in hosts:
 86                # Всеки домакин обслужва децата при него
 87                for kid in kids:
 88                    kid_position = kid.get_position()
 89                    host_position = host.get_position()
 90                    distance = abs(kid_position[0] - host_position[0]) + abs(kid_position[1] - host_position[1])
 91                    if distance == 0:
 92                        # Детето е при домакина
 93                        host.remove_candy(kids.remove(kid))
 94                    else:
 95                        # Детето се движи към най-близкия домакин, който още не е навестено
 96                        if not kid_position:
 97                            new_position = host_position
 98                        else:
 99                            if distance < abs(kid_position[0] - host_position[0]) + abs(kid_position[1] - host_position[1]):
100                                new_position = host_position
101                            else:
102                                new_position = kid_position
103
104                        kid.set_position(new_position)
105                        new_participants.add(kid)
106
107            # Обновяване на списъка с участниците
108            self.participants = new_participants.union(hosts)
109
110            # Проверка за края на симулацията
111            if not kids:
112                return None   
113            
114                     
115candy = Candy(20, 0.3)
116person = Person((1, 2))
117kid = Kid((0, 0), 123)
118host = Host((3, 4), [(1, 1.0), (2, 0.5)])
119flux_capacitor = FluxCapacitor({kid, host})