解题思路:(a+b+c)%k等价于(a%k+b%k+c%k)%k,之所以这样是因为直接暴力a,b,c范围太大超时,采用余数的方法可以减少时间复杂度,对于每个余数
我们只用找到对应三个最大的数就够用了。
注意事项:
参考代码:
from sys import maxsize
n, k = map(int, input().split())
r = [[-maxsize] * 3 for _ in range(k)]
a = [int(i) for i in input().split()]
for i in range(n):
re = a[i] % k
if a[i] > r[re][0]:
r[re][0], r[re][1], r[re][2] = a[i], r[re][0], r[re][1]
elif a[i] > r[re][1]:
r[re][1], r[re][2] = a[i], r[re][1]
elif a[i] > r[re][2]:
r[re][2] = a[i]
Max = -maxsize
for i in range(k):
for j in range(k):
tmp = (k - i - j) % k # 第三个余数
if i == j:
if tmp == i:
Max = max(Max, r[i][0] + r[i][1] + r[i][2])
else:
Max = max(Max, r[i][0] + r[i][1] + r[tmp][0])
elif j == tmp:
Max = max(Max, r[i][0] + r[j][0] + r[j][1])
else:
if i == tmp:
Max = max(Max, r[i][0] + r[i][1] + r[j][0])
else:
Max = max(Max, r[i][0] + r[j][0] + r[tmp][0])
print(Max)
0.0分
0 人评分