In this Maximize It! problem we need to develop a python program that can read an integer input separated with lines and then we need to print the maximum value on the output screen.
Problem solution in Python 2 programming.
# Enter your code here. Read input from STDIN. Print output to STDOUT K, M = map(int, raw_input().strip().split(' ')) ls = [] for i in range(K): l = map(int, raw_input().strip().split(' ')[1:]) ls.append(l) level = len(ls) - 1 res = 0 def loopf(ls, level, s, m): global res if level < 0: if s > res: res = s return for i in ls[level]: loopf(ls, level-1, (s + i ** 2 % m) % m, m) loopf(ls, level, 0, M) print res
Problem solution in Python 3 programming.
# Enter your code here. Read input from STDIN. Print output to STDOUT from itertools import product K,M = map(int,input().split()) N = (list(map(int, input().split()))[1:] for _ in range(K)) results = map(lambda x: sum(i**2 for i in x)%M, product(*N)) print(max(results))
Problem solution in pypy programming.
import itertools (K, N) = map(int, raw_input().split()) L = list() for i in range(K): l = map(int, raw_input().split()) n = l[0] L.append(l[1:]) assert len(L[i]) == n S_max = 0 L_max = None for l in itertools.product(*L): s = sum([x**2 for x in l]) % N if s > S_max: S_max = s L_max = l print S_max
Problem solution in pypy3 programming.
# Enter your code here. Read input from STDIN. Print output to STDOUT from itertools import product args = list(map(int, input().split())) print(max([sum([y**2 for y in x])%args[1] for x in product(*[map(int,input().split()[1:]) for _ in range(args[0])])]))