goods, где goods[i] = [profit, category] — выручка от товара и его категория. На витрину нужно поставить ровно count товаров (каждый товар из списка можно взять не больше одного раза).
Ценность витрины равна сумме выручки выбранных товаров плюс **квадрат** числа различных категорий среди них.
Верните наибольшую возможную ценность витрины.goods=[[7, 1], [6, 1], [2, 2]], count=2 → 14
Два товара категории 1 дают 7 + 6 + 1² = 14. Товары [7, 1] и [2, 2] дают 7 + 2 + 2² = 13. Лучше 14.
goods=[[10, 3], [9, 3], [1, 1], [1, 2]], count=3 → 24
Берём [10, 3], [9, 3] и [1, 1]: выручка 20, две категории, 20 + 2² = 24. Три разные категории дают лишь 10 + 1 + 1 + 3² = 21.
goods=[[5, 1]], count=1 → 6
goods=[[7, 1], [6, 1], [2, 2]], count=2
14
«Запустить» — первые 3 теста у тебя в браузере, «Отправить» — все 108 тестов на сервере, включая скрытые.