Язык не указан, напишу на Python'е :)
n = int(input())
s = 0
for i in range(n):
s += n**2
print(s)
250,140625 килобайт
если нужно решение то вот:
i=64
K=128*256
I=?
Так как в коробке всего 31 шар, а дано 12 красных и 17 желтых, то мы можем найти количество шаров, цвет которых мы точно знаем(12+17=29). 29 меньше, чем 31, значит, цвет еще двух шаров мы узнать не можем, т.е они могут быть любого цвета. И, так как 29<31, ошибки в условии задачи нет(цвет двух оставшихся шаров просто не дан)