每日一题：摘桃子

 摘桃子 - 题目 - 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;
}