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.uranium * self.mass
 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, position):
 26        self.position = position
 27
 28
 29class Kid(Person):
 30
 31    CRITICAL_QUANTITY = 20
 32
 33    def __init__(self, position, initiative):
 34        super().__init__(position)
 35        self.initiative = initiative
 36        self.basket = []
 37        self.visited_hosts = set()
 38
 39    def get_initiative(self):
 40        return self.initiative
 41
 42    def add_candy(self, candy):
 43        self.basket.append(candy)
 44
 45    def is_critical(self):
 46        sum = 0
 47        for candy in self.basket:
 48            sum += candy.get_uranium_quantity()
 49        return sum > self.CRITICAL_QUANTITY
 50
 51    def visit_closest_host(self, hosts):
 52        if len(self.visited_hosts) == len(hosts):
 53            return
 54
 55        closest_host = None
 56        for host in hosts:
 57            if host in self.visited_hosts:
 58                continue
 59            if closest_host is None:
 60                closest_host = host
 61            if math.dist(self.position, host.get_position()) < math.dist(self.position, closest_host.get_position()):
 62                closest_host = host
 63            elif math.dist(self.position, host.get_position()) == math.dist(self.position, closest_host.get_position()):
 64                if (host.get_position()[0] < closest_host.get_position()[0]
 65                    or host.get_position()[1] < closest_host.get_position()[1]):
 66                    closest_host = host
 67        
 68        closest_host.visited_by_kid(self)
 69        self.visited_hosts.add(closest_host)
 70        self.position = closest_host.get_position()
 71
 72
 73class Host(Person):
 74
 75    def __init__(self, position, candies):
 76        super().__init__(position)
 77        self.visitors = []
 78
 79        self.candies = set()
 80        for pair in candies:
 81            candy = Candy(pair[0], pair[1])
 82            self.candies.add(candy)
 83
 84    def remove_candy(self, remove_by_condition):
 85        if len(self.candies) == 0:
 86            return None
 87        removed = remove_by_condition(self.candies)
 88        self.candies.discard(removed)
 89        return removed
 90
 91    def visited_by_kid(self, kid):
 92        self.visitors.append(kid)
 93
 94    def treat(self):
 95        self.visitors.sort(key=lambda x: x.get_initiative(), reverse=True)
 96        for visitor in self.visitors:
 97            candy = self.remove_candy(removal_condition)
 98            if candy:
 99                visitor.add_candy(candy)
100        self.visitors.clear()
101
102
103class FluxCapacitor:
104
105    def __init__(self, participants):
106        self.participants = participants
107        self.hosts = [participant for participant in self.participants if isinstance(participant, Host)]
108        self.kids = [participant for participant in self.participants if isinstance(participant, Kid)]
109
110    def get_victim(self):
111        visitors = len(self.kids)
112
113        while visitors > 0:
114            visitors = len(self.kids)
115            for kid in self.kids:
116                kid.visit_closest_host(self.hosts)
117                visitors -= len(kid.visited_hosts) == len(self.hosts)
118
119            for host in self.hosts:
120                host.treat()
121
122            critical_kids = set()
123            for kid in self.kids:
124                if kid.is_critical():
125                    critical_kids.add(kid)
126
127            if len(critical_kids) > 0:
128                return critical_kids
129
130        return None
131
132
133def removal_condition(candies):
134    largest_candy = list(candies)[0]
135    for candy in candies:
136        if candy.get_mass() > largest_candy.get_mass():
137            largest_candy = candy
138    
139    return largest_candy