1class Candy:
  2    def __init__(self, mass, uranium):
  3        self.mass = mass
  4        self.uranium = uranium
  5
  6    def get_uranium_quantity(self):
  7        uranium_quantity = self.mass * self.uranium
  8        return uranium_quantity
  9
 10    def get_mass(self):
 11        return self.mass
 12
 13class Person:
 14    def __init__(self, position):
 15        self.position = position
 16
 17    def get_position(self):
 18        return self.position
 19
 20    def set_position(self, new_position):
 21        self.position = new_position
 22
 23
 24class Kid(Person):
 25    def __init__(self, position, initiative):
 26        super().__init__(position)
 27        self.initiative = initiative
 28        self.candy_basket = []
 29
 30    def get_initiative(self):
 31        return self.initiative
 32
 33    def add_candy(self, candy_instance):
 34        self.candy_basket.append(candy_instance)
 35
 36    def is_critical(self):
 37        total_uranium = sum(candy.get_uranium_quantity() for candy in self.candy_basket)
 38
 39        return total_uranium > 20
 40
 41
 42class Host(Person):
 43    def __init__(self, position, candies):
 44        super().__init__(position)
 45        self.candy_basket = []
 46
 47        for candy_args in candies:
 48            candy_instance = Candy(*candy_args)
 49            self.candy_basket.append(candy_instance)
 50
 51    def remove_candy(self, selection_function):
 52        if not self.candy_basket:
 53            return None
 54        else:
 55            selected_candy = selection_function(self.candy_basket)
 56
 57            self.candy_basket.remove(selected_candy)
 58
 59            return selected_candy
 60
 61def max_candy_mass(candies):
 62    return max(candies, key=lambda candy: candy.get_mass())
 63
 64class FluxCapacitor:
 65    def __init__(self, participants):
 66        self.participants = participants
 67
 68    def get_victim(self):
 69        while True:
 70            next_round_kids = set() 
 71            victims = set()
 72            all_visited_hosts = set()
 73
 74            for kid in self.participants:
 75                if isinstance(kid, Kid):
 76                    if kid in all_visited_hosts:
 77                        continue
 78
 79                    closest_host = None
 80                    closest_distance = float('inf')
 81
 82                    for host in self.participants:
 83                        if isinstance(host, Host) and host not in all_visited_hosts:
 84                            distance = abs(kid.get_position()[0] - host.get_position()[0]) + abs(kid.get_position()[1] - host.get_position()[1])
 85                            if distance < closest_distance:
 86                                closest_host = host
 87                                closest_distance = distance
 88
 89                    if closest_host is not None:
 90                        candy = closest_host.remove_candy(max_candy_mass)
 91                        if candy:
 92                            kid.add_candy(candy)
 93                            if kid.is_critical():
 94                                victims.add(kid)
 95                        all_visited_hosts.add(closest_host)
 96                        next_round_kids.add(kid)
 97                    else:
 98                        all_visited_hosts = set(self.participants)
 99
100            if not next_round_kids:
101                return None
102
103            if victims:
104                return victims
105            else:
106                self.participants = next_round_kids
107
108
109candy = Candy(20, 0.3)
110person = Person((1, 2))
111kid = Kid((0, 0), 123)
112host = Host((3, 4), [(1, 1.0), (2, 0.5)])
113flux_capacitor = FluxCapacitor({kid, host})
114
115victims = flux_capacitor.get_victim()
116
117if victims:
118    print("Victims found:")
119    for victim in victims:
120        print(victim.get_position())
121else:
122    print("No victims found.")