-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathItertools.py
More file actions
64 lines (58 loc) · 1.69 KB
/
Itertools.py
File metadata and controls
64 lines (58 loc) · 1.69 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
##itertools.product()
# from itertools import product
# A = map(int, '1 2'.split(' '))
# B = map(int, '3 4'.split(' '))
# print(*product(A, B))
##itertools.permutations()
# from itertools import permutations
# s,k = input().split()
# l=[]
# for line in tuple(permutations(sorted(s),int(k))):
# l.append(''.join(line))
# print(l)
# for line in l:
# print(line)
##itertools.combinations()
from itertools import combinations
# s,k = input().split()
#print('\n'.join([str(line) for line in sorted(s)]))
# for i in range(1,int(k)+1):
# print(*[''.join(line) for line in combinations(sorted(s),int(i))],sep='\n')
#itertools.combinations_with_replacement()
#from itertools import combinations_with_replacement
#s = 'HACK'
#k = '2'
#print(*[''.join(line) for line in combinations_with_replacement(sorted(s),int(k))],sep='\n')
#Compress the String!
# from itertools import groupby
# s = '1222311'
# print(*[(len(list(g)),int(k)) for k, g in groupby(s)])
# print()
# for k, g in groupby(s):
# l = []
# l.append(len(list(g)))
# l.append(int(k))
# print(tuple(l),end=' ')
# print()
##Iterables and Iterators
# from itertools import combinations
# l = []
# m = []
# l = [''.join(i) for i in combinations('aacd',2) if i.count('a')]
# m = [''.join(i) for i in combinations('aacd',2) if not i.count('a')]
# print('{0:.3}'.format(len(l)/(len(l)+len(m))))
##Maximize It!
from itertools import product
f = open('testcase_0.txt','r')
k, m = map(int, f.readline().split())
l=[]
for _ in f:
l.append(list(map(int, _.split(' ')))[1:])
results = map(lambda x: sum(i**2 for i in x)%m, product(*l))
print(max(results))
# l=[]
# for line in x:
# l.append(max(line)**2)
# print(l)
# print()
# print(abs(sum(l)%m))