题目描述
众所周知,小葱同学擅长计算,尤其擅长计算一个数是否是另外一个数的倍数。但小葱只擅长两个数的情况,当有很多个数之后就会比较苦恼。现在小葱给了你 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)