Maximize It!

  • + 0 comments
    1. k , m = map(int,input().split())
    2. li = []
    3. dp = set()
    4. for i in range (k):
    5. x = list(map(int,input().split()))
    6. li.append(x)
    7. for idx,j in enumerate(li[0]):
    8. if(idx!=0):
    9. dp.add((j**2)%m)
    10. for l in range(1,k):
    11. new_dp = set()
    12. for ind , z in enumerate(li[l]):
    13. if(ind!=0):
    14. n_s_m = (z**2)%m
    15. for ele in dp:
    16. new_dp.add((n_s_m + ele)%m)
    17. dp = new_dp
      1. print(max(dp))