蓝桥杯 K倍区间 python组

题目描述

给定一个长度为 N 的数列，A1​,A2​,⋯AN​，如果其中一段连续的子序列Ai​,Ai​+1,⋯Aj​ (i≤j ) 之和是 K 的倍数，我们就称这个区间 [i, j]是 K 倍区间。

你能求出数列中总共有多少个 K 倍区间吗？

输入描述

第一行包含两个整数 N 和 K( 1≤N,K≤10**5 )。

以下 N 行每行包含一个整数Ai​ ( 1≤Ai​≤10**5 )

输出描述

输出一个整数，代表 K 倍区间的数目。

解答：

最开始我是通过从每一个值开始，往后递增后面的值，对每一次的和进行计算，如果对K余数为零，则K倍区间的个数增加一，即暴力法，但是超时，10**10明显是大于要求的时间2乘以10**9。

所以只能换种思路，因为每一个数都可以加上前面的数的值，存放在该位置上，即前缀和的相关知识点：

1，前缀和对K取余，为0则数目加一；

2，两个余数相同的前缀和，他们的差对K取余一定为一种方法，例如3%2=1，5%2=1，所以（5-3）%2=0。

3，多个余数相同的前缀和，方案数也会增加，比如3，5，7会有（5-3）、（7-3）、（7-5）三中情况，也就是1+2=3，在ans加的过程中就是先加1，在加2（看代码中的ans那里）。

1和2均满足第3种情况，可以说是3是对1和2的一种代码解析方式。

n,k = map(int,input().split())
'''
ans = []
for i in range(n):
  ans.append(int(input()))

count = 0
for i in range(len(ans)):
  if ans[i]%k==0: 
    count+=1
  if ans[i]==(len(ans)-1):
    break
  num = ans[i]
  for j in range(i+1,len(ans)):
    num += ans[j]
    if num%k==0:
      count+=1

print(count)'''
dp=[0 for i in range(k)]    #创建一个对K取余的所有余数的数组，并把初值全部赋为一
dp[0]=1    #第一次当余数为0时是倍数
ans=0#计数
p=[0]    #存放输入的数据的前缀和
for i in range(n):
    tep=int(input())
    p.append(p[-1]+tep)    #添加的数为输入的数加上之前数的总和，即为前缀和
    m=p[i+1]%k    #求余数
    ans+=dp[m]    
    #计数作用，三种符合条件的情况全部计算进去，如果dp[m]的值有很多则个数也要增加，如果dp[m]的值为三，则三个元素的差值会有1+2=3中组合
    dp[m]+=1    #余数为m的位置个数加一

print(ans)