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, position):
 26        self.position = position
 27
 28
 29class Kid(Person):
 30
 31    def __init__(self, position, initiative):
 32        super().__init__(position)
 33        self.initiative = initiative
 34        self._basket = []
 35        self._visited_hosts = []
 36
 37    def get_initiative(self):
 38        return self.initiative 
 39    
 40    def add_candy(self, candy):
 41        self._basket.append(candy)
 42
 43    def is_critical(self):
 44        total_uranium = 0.0
 45        for candy in self._basket:
 46            total_uranium += candy.get_uranium_quantity()
 47        return total_uranium > FluxCapacitor.URANIUM_LIMIT
 48      
 49    def visit_host(self,hosts):
 50            for host in sorted(hosts, key=lambda x : (math.dist(self.position,x.get_position()), x.position[0], x.position[1])):
 51                if host in self._visited_hosts:
 52                    continue
 53                else:
 54                    candy = host.remove_candy(FluxCapacitor.choose_candy)
 55                    self.set_position(host.get_position())
 56                    if candy is not None:
 57                        self._basket.append(candy)
 58                    self._visited_hosts.append(host)
 59                    return True
 60            return False
 61
 62
 63class Host(Person):
 64
 65    def __init__(self, position, candies):
 66        super().__init__(position)
 67        self.candies = []
 68        for candy in candies:
 69            self.candies.append(Candy(candy[0], candy[1]))
 70
 71    def remove_candy(self, remove_func):
 72        if not self.candies:
 73            return None
 74        else:
 75            candy = remove_func(self.candies)
 76            self.candies.remove(candy)
 77            return candy
 78
 79
 80class FluxCapacitor:
 81
 82    URANIUM_LIMIT = 20
 83
 84    def __init__(self, participants):
 85        self.participants = participants
 86
 87    @staticmethod
 88    def choose_candy(candies):
 89        return max(candies, key=lambda x : x.get_mass())
 90
 91    def get_victim(self):
 92        dead_kids = []
 93        kids = sorted([kid for kid in self.participants if type(kid) is Kid], key=lambda x: x.initiative, reverse=True)
 94        hosts = [host for host in self.participants if type(host) is Host]
 95        while not dead_kids and kids:
 96            for kid in kids:
 97                if kid.visit_host(hosts):
 98                    if kid.is_critical():
 99                        dead_kids.append(kid)
100                else:
101                    kids.remove(kid)
102        if dead_kids:
103            return dead_kids
104        else:
105            return None