1import math
  2
  3
  4class Candy:
  5
  6    def __init__(self, mass, uranium):
  7        self.mass = mass
  8        self.uranium = uranium
  9
 10    def get_uranium_quantity(self):
 11        return self.mass * self.uranium
 12    
 13    def get_mass(self):
 14        return self.mass
 15
 16
 17class Person:
 18
 19    def __init__(self, position):
 20        self.position = position
 21
 22    def get_position(self):
 23        return self.position
 24    
 25    def set_position(self, new_position):
 26        self.position = new_position
 27
 28
 29class Kid(Person):
 30
 31    LETHAL_DOSE = 20
 32
 33    def __init__(self, position, initiative):
 34        super().__init__(position)
 35        self.initiative = initiative
 36        # we will store the kid`s candy in a list named basket
 37        self.basket = []
 38        self.visited = []
 39
 40    def get_initiative(self):
 41        return self.initiative
 42    
 43    def add_candy(self, candy):
 44        self.basket.append(candy)
 45
 46    def is_critical(self):
 47        total_uranium_quantity = 0
 48        for candy_item in self.basket:
 49            total_uranium_quantity += candy_item.get_uranium_quantity()
 50
 51        return total_uranium_quantity > self.LETHAL_DOSE
 52
 53
 54class Host(Person):
 55    
 56    def __init__(self, position, candies):
 57        super().__init__(position)
 58        self.candies = [Candy(mass, uranium) for mass, uranium in candies]
 59
 60    def choose_candy(self, candies):
 61        return max(candies, key=lambda candy: candy.get_mass())
 62        
 63    def remove_candy(self, choose_candy_func):
 64        if len(self.candies) == 0:
 65            return None
 66        
 67        choosen_candy = choose_candy_func(self.candies)
 68        self.candies.remove(choosen_candy)
 69        return choosen_candy
 70
 71
 72class FluxCapacitor:
 73
 74    def __init__(self, participants):
 75        self.participants = participants
 76        self.hosts = []
 77        self.kids = []
 78        for participant in self.participants:
 79            if isinstance(participant, Host):
 80                self.hosts.append(participant)
 81            elif isinstance(participant, Kid):
 82                self.kids.append(participant)
 83
 84        self.kids = sorted(self.kids, key=lambda kid: kid.get_initiative())
 85        
 86    def get_victim(self):
 87        dead_kids = []
 88        is_dead = False
 89        while not is_dead:
 90            kid: Kid
 91            for kid in self.kids:
 92                current_host = None
 93                current_dist = float('inf')
 94                host: Host
 95                for host in self.hosts:
 96                    if host in kid.visited:
 97                        continue
 98                    distance = math.dist(kid.get_position(), host.get_position())
 99                    if distance < current_dist:
100                        current_dist = distance
101                        current_host = host
102                    elif current_dist == distance and host.get_position() < current_host.get_position():
103                        current_dist = distance
104                        current_host = host
105
106                if current_host == None:
107                    is_dead = True
108                    break
109
110                kid.visited.append(current_host)
111                picked_candy = current_host.remove_candy(host.choose_candy)
112
113                if picked_candy is not None:
114                    kid.basket.append(picked_candy)
115
116                    if kid.is_critical():
117                        is_dead = True
118                        dead_kids.append(kid)
119
120        return dead_kids or None