2018蓝桥杯真题：倍数问题python

题目描述

众所周知，小葱同学擅长计算，尤其擅长计算一个数是否是另外一个数的倍数。但小葱只擅长两个数的情况，当有很多个数之后就会比较苦恼。现在小葱给了你 n 个数，希望你从这 n 个数中找到三个数，使得这三个数的和是 K 的倍数，且这个和最大。数据保证一定有解。

输入描述

第一行包括 2 个正整数 n, K。

第二行 n 个正整数，代表给定的 n 个数。 

其中，1≤n ≤10^5, 1≤K ≤10^3，给定的 n 个数均不超过 10^8。

输出描述

输出一行一个整数代表所求的和。

输入输出样例

示例

输入

4 3
1 2 3 4

输出

9

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

写if语句的时候，一定要思考清楚，注意if和else搭配

代码：（通过测试）

n,k=map(int,input().split())
l=list(map(int,input().split()))
fenlei=[[0,0,0] for i in range(k)]
for i in range(n):
      if fenlei[l[i]%k][0]==0:
            fenlei[l[i]%k][0]=l[i]
      elif fenlei[l[i]%k][1]==0:
            fenlei[l[i]%k][1]=l[i]
      elif fenlei[l[i]%k][2]==0:
            fenlei[l[i]%k][2]=l[i]
      else:
            fenlei[l[i]%k].append(l[i])
for i in range(k):
      fenlei[i].sort()
ans=[0 for i in range(3)]
result=0
for i in range(k):
      for j in range(i,k):
            m=(k-(i+j)%k)%k
           
            ans[0]=fenlei[i][-1]
            if i==j:
                  ans[1]=fenlei[i][-2]
                  
                  if i==m:
                        ans[2]=fenlei[i][-3]
                  else:
                        ans[2]=fenlei[m][-1]
            else:
                  ans[1]=fenlei[j][-1]
                  if i==m or j==m:
                        ans[2]=fenlei[m][-2]
                  else:ans[2]=fenlei[m][-1]
            if sum(ans)>result:
                  result=sum(ans)
print(result)