Practice>Python>Itertools>Maximize it!

[jmpark0118]

출처 : https://www.hackerrank.com/challenges/maximize-it/problem

[jmpark0118]

from itertools import product

if __name__ == '__main__':
    k, m = map(int, input().split())

    my_list = (list(map(int, input().split()))[1:] for _ in range(k))
    answer = map(lambda x: sum(i**2 for i in x)%m, product(*my_list))

    print(max(list(answer)))

[jmpark0118]

각 줄에서 가장 큰 숫자를 찾는 것이 아니라 나머지의 합이 최대화 되도록 하는 숫자를 찾는 문제!!