# Шаблон кода:
import fileinput
import json
# дополнительный импорт библиотек, если требуются
# ...
# объявление функций
def get_friend(x):
# ваш код
return result # вывод результата работы вашей функции
# считывание из stdin
for line in fileinput.input():
# преобразование в объект python
candies = json.loads(line)
# вызов функции и вывод результата в консоль
print(get_friend(candies))