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
20    def get_position(self):
21        return self.position
22
23    def set_position(self, new_position):
24        self.position = new_position
25
26
27class Kid(Person):
28    def __init__(self, position, initiative):
29        super().__init__(position)
30        self.initiative = initiative
31        self.candies = []
32
33    def get_initiative(self):
34        return self.initiative
35
36    def add_candy(self, candy):
37        self.candies.append(candy)
38
39    def is_critical(self):
40        uranium_quantity = sum(candy.get_uranium_quantity() for candy in self.candies)
41        return uranium_quantity > 20
42
43
44class Host(Person):
45    def __init__(self, position, candies):
46        super().__init__(position)
47        self.candies = [Candy(mass, uranium) for mass, uranium in candies]
48    
49    def remove_candy(self, choose_candy):
50        if not self.candies:
51            return None
52        candy_to_remove = choose_candy(self.candies)
53        self.candies.remove(candy_to_remove)
54        return candy_to_remove
55
56
57class FluxCapacitor:
58    def __init__(self, participants):
59        self.participants = participants
60    
61    def get_victim(self):
62        kids = [participant for participant in self.participants if isinstance(participant, Kid)]
63        hosts = [participant for participant in self.participants if isinstance(participant, Host)]
64        critical_kids = set()
65        while hosts and kids:
66            for kid in sorted(kids, key=lambda kid: kid.get_initiative(), reverse=True):
67                min_distance = float('inf')
68                closest_host = None
69                for host in hosts:
70                    distance = math.sqrt((host.get_position()[0] - kid.get_position()[0])**2 + (host.get_position()[1] - kid.get_position()[1])**2)
71                    if distance < min_distance:
72                        min_distance = distance
73                        closest_host = host
74                if closest_host:
75                    selected_candy = closest_host.remove_candy(lambda candies: max(candies, key=lambda candy: candy.get_mass()))
76                    if selected_candy:
77                        kid.add_candy(selected_candy)
78                        kid.set_position(closest_host.get_position())
79                        if kid.is_critical():
80                            critical_kids.add(kid)
81                    hosts.remove(closest_host)
82            if not hosts:
83                break  
84        return critical_kids if critical_kids else None