We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Prepare
  2. Python
  3. Itertools
  4. Maximize It!
  5. Discussions

Maximize It!

  • jmauriciomejia1
     + 0 comments

    My solution (not very elegant, but passed all tests):

    from itertools import product
from functools import partial

k, m = map(int, input().split())

def s(lst: tuple, mod: int) -> int:
    return sum(map(lambda x: x**2, lst)) % mod
    
s_fixed = partial(s, mod=m)
lsts = []

for _ in range(k):
    n, *arr = map(int, input().split())
    lsts.append(arr)

print(max(map(s_fixed, product(*lsts))))