题目描述:
已知一个数组中的所有数据,已知数组的长度为n,已知一个除数为k。要求我们从这组数据中找到这样的区间,其中所有的元素之和能够被k整除。最终统计出所有的区间数目
输入描述:
第一行:
n,k表示数组的长度,和除数k
第二行及后面的n 1行:
每行一个数字,表示初始数组中的所有元素的数据
输出描述:
输出满足上述条件(区间中所有元素的和可以整除k)所有区间数
样例输入输出:
样例输入:
5 2
1
2
3
4
5
样例输出:
6
算法:
最开始暴力解法,当然是没有通过所有案例
代码语言:javascript复制import os
import sys
n,k = map(int,input().split())
ans = []
for i in range(n):
a = int(input())
ans.append(a)
cnt = 0
for l in range(1,n 1):
for i in range(n-l 1):
res = sum(ans[i:i l])
if res%k==0:
cnt =1
print(cnt)
利用前缀和的思想,和一些数学的思维的AC代码
代码语言:javascript复制import os
import sys
#前缀和、优化
n,k = map(int,input().split())
cnt = [0]*(k)
ans = 0
for i in range(n):
a = int(input())
ans =a
cnt[ans%k] =1
##print(cnt[0],cnt[1])
ans = cnt[0]
for i in range(k):
ans =(cnt[i]*(cnt[i]-1))//2
print(ans)