摘桃子 - 题目 - Daimayuan Online Judge
用sum[i]求前缀和(n小于等于2e5,ai小于等于1e9,所以前缀和最大2e14,超出int范围了,应该用long long),该题问的就是有多少个子段[l,r]满足(sum[r]-sum[l-1])%k==r-l+1(说明r-l+1<k)
进行等价变形:
举个例子:5%2==1,则5=2*m+1(其中m==5/2,总之m是个整数)
同理,sum[r]-sum[l-1]=k*m+r-l+1(其中m==(sum[r]-sum[l-1]/k,总之m是个整数)
得sum[r]-r==k*m+sum[l-1]-(l-1)(其中m是个整数,但不一定是(sum[r]-r)/k,m是要比(sum[r]-r)/k小的,也就是说sum[l-1]-(l-1)还可以分解为k*n+r,其中m+n==(sum[r]-r)/k,(sum[l-1]-(l-1)%k==r)
所以(sum[r]-r)%k==(sum[l-1]-(l-1))%k(其中sum[r]-r肯定大于等于0,sum[l-1]-(l-1)也肯定大于等于0,因为ai>=1,所以1到i的前缀和一定大于等于从1到i的长度的)
这样问题就转换成有多少个子段[l,r]满足(sum[r]-r)%k==(sum[l-1]-(l-1))%k,且要保证r-l+1小于k(区间长度小于k)
可以从1到n遍历,对于每一个(sum[i]-i)%k的值,用map记录该值的个数,如果后面再次出现这个值,那么就乘以前面该值已经出现过的个数,就是用乘法原理1*前面出现过的个数来算有多少个这样的子段,但是因为子段长度要小于k,所以当i大于等于k时,i每加一,就要减去满足的长度范围之外的那个值的个数
mp[0]一开始赋值为1,原因是条件是(sum[r]-r)%k==(sum[l-1]-(l-1))%k,注意是l-1,而不是l,l从1到n,l等于1时,(sum[0]-0)%k==0,一开始mp[0]就是1
当l==n,r==n时,如果满足条件的话,就说明这一个数字满足%k==1
AC代码:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
using namespace std;
typedef long long LL;
map<int,int>mp;
const int N = 200010;
LL a[N], sum[N];
int main()
{
int n, k;
cin >> n >> k;
for (int i = 1; i <= n; i++) {
cin >> a[i];
sum[i] = sum[i - 1] + a[i];
}
mp[0]++;
LL ans = 0;
for (int i = 1; i <= n; i++) {
if (i >= k) mp[(sum[i - k] - (i - k)) % k]--;
ans+=mp[(sum[i] - i) % k];
mp[(sum[i] - i) % k]++;
}
cout << ans << endl;
return 0;
}