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!

Sort by

recency

|

1108 Discussions

|

  • dilajevolves
     + 0 comments
    from itertools import product
K,M = map(int,input().split())
result_list = [list(map(int,input().split()[1:])) for _ in range(K)]
print(max(sum(x**2 for x in combi)%M for combi in product(*result_list)))

  • maxence_qnt
     + 1 comment

    After several try, I finally pass this Hard problem with the following code. It's maybe possible to improve it in term of time complexity. The logic is to take all product possibilities using the itertool.product() method, then, with a temp value, see if the current result is better than the output.

    # Enter your code here. Read input from STDIN. Print output to STDOUT
from itertools import product

K, M = map(int, input().split())

all_lists = []

for i in range(K):
    all_lists.append(list(map(int, input().split()[1:])))

#print(all_lists)

for sub_list in all_lists:
    sub_list.sort() 



#all_product_possibilities = list(product(*all_lists))
# print(all_product_possibilities[:5])

output = 0

#print("-------------------------")

for possible_product in list(product(*all_lists)):
    current_product = list(possible_product)
    #print(f"current_product = {current_product}")
    temp = 0
    for values in current_product:
        temp += values**2
    
    temp = temp%M
    #print(f"Temp: {temp}")
    #print(f"Output: {output}")
    if temp > output:
        output = temp
    else:
        continue
    #print(f"Current Output = {output}")
print(output)

  • rohitambasta_ii1
     + 0 comments
    from itertools import product

K, M= list(map(int, input().split()))
inp= [list(map(int, input().split()))[1::] for _ in range(K)]

super_squared_elements = []
for i in inp:
    squared_elements = []
    for j in i:
        squared_elements.append(j*j)
    super_squared_elements.append(squared_elements)
  
cartesian_product= list(product(*super_squared_elements))

max_check_appended= []

for val in cartesian_product:
    max_check_appended.append(sum(val) % M)

print(max(max_check_appended))

  • napalmcattt
     + 0 comments
    from itertools import product

K, M = map(int, input().split())
a = [list(map(int, input().split()))[1:] for _ in range(K)]

print(max([sum(map(lambda x: x**2, combo)) % M for combo in product(*a)]))

  • nehaasharma_22_1
     + 0 comments
    from itertools import product

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

    arrays = []

    for _ in range(1, k + 1):
        arr = list(input().strip().split())[1:]
        arrays.append(arr)

    s_max = 0

    for combo in product(*arrays):
        sum_combo = sum(int(x) ** 2 for x in combo)
        current_s_max = sum_combo % n
        s_max = max(current_s_max, s_max)

    print(s_max)