1class Candy:
  2
  3    def __init__(self, mass, uranium):
  4        self.mass = mass
  5        self.uranium = uranium
  6
  7    def get_uranium_quantity(self):
  8        return self.mass * self.uranium
  9
 10    def get_mass(self):
 11        return self.mass
 12
 13class Person:
 14
 15    def __init__(self, position):
 16        self.position = position # tuple of ints
 17
 18    def get_position(self):
 19        return self.position
 20
 21    def set_position(self, new_position):
 22        self.position = new_position
 23
 24class Kid(Person):
 25
 26    THRESHOLD = 20
 27
 28    def __init__(self, position, initiative):
 29        Person.__init__(self, position)
 30        self.initiative = initiative
 31        self.candy_arr = [] # here I will gather the kids candies
 32
 33    def get_initiative(self):
 34        return self.initiative
 35
 36    def add_candy(self, candy): # put a candy in the candy_arr
 37        self.candy_arr.append(candy)
 38
 39    def is_critical(self): 
 40        return sum(candy.get_uranium_quantity() for candy in self.candy_arr) > Kid.THRESHOLD
 41
 42    @staticmethod
 43    def __sort_hosts(host):
 44        x, y = host.get_position()
 45        return (x, y)
 46
 47    def find_nearest_host(self, hosts):
 48        if not hosts:
 49            return None
 50        else:
 51            sorted_hosts = sorted(hosts, key=self.__sort_hosts) # min func can be used
 52            nearest_host = sorted_hosts[0]
 53
 54        return nearest_host
 55
 56class Host(Person):
 57
 58    def __init__(self, position, candies):
 59        Person.__init__(self, position)
 60        self.candy_arr = [Candy(mass, uranium) for mass, uranium in candies] # list of tupes of two elements ( int, float )
 61
 62    def remove_candy(self, func):
 63        if not self.candy_arr:
 64            return None
 65
 66        chosen_candy = func(self.candy_arr)
 67        self.candy_arr.remove(chosen_candy)
 68        return chosen_candy
 69
 70    def candy_max_mass(candies): # max mass
 71
 72        def get_mass(candy):
 73            return candy.get_mass()
 74
 75        return max(candies, key=get_mass)
 76
 77class FluxCapacitor:
 78
 79    def __init__(self, participants):
 80        self.participants = participants # set of Kid and Host instances
 81
 82    @staticmethod
 83    def __sort_kids_by_initiative(kid):
 84        return kid.get_initiative()
 85
 86    def get_victim(self):
 87        victims = set()
 88
 89        kids = set()
 90        hosts = set()
 91        for item in self.participants:
 92            if type(item) == Kid:
 93                kids.add(item)
 94            elif type(item) == Host:
 95                hosts.add(item)
 96
 97        sorted_kids = sorted(kids, key=self.__sort_kids_by_initiative, reverse=True)
 98
 99        for kid in sorted_kids:
100            nearest_host = kid.find_nearest_host([host for host in hosts])
101            if nearest_host:
102                for candy in nearest_host.candy_arr:
103                    kid.add_candy(candy)
104
105        for kid in sorted_kids:
106            if kid.is_critical(): # kid is critical -> true or false, < or > 20
107                victims.add(kid)
108
109        if victims:
110            return victims
111        return None