摘桃子(前缀和)

                                                            摘桃子        

摘桃子 - 题目 - Daimayuan Online Judge 

思路：因为是连续的一段，所以我们还是用前缀和的思想考虑，假设sum[i]为前i个数的和，那么我们需要找到一对l,r使得(sum[r]-sum[l-1])%k=r-l+1，通过整理我们能够得到

(sum[r]-r)%k=(sum[l-1]-(l-1))%k，（为什么能得到这个式子呢， 因为要让等式成立，则需要满足r-r-l+1<k，则(r-l+1)%k==r-l+1的，所以我们同模k是相等的），所以我们可以从左到右遍历，用一个map维护(sum[i]-i)%k的值，每次我们加上这个值即可，需要注意的是st[0]需要为1，因为其实它可以选择l=0作为左边界，所以我们需要加上，另外有一个问题就是我们在进行撤销判断时用的时i>=k，那么假设是n=3 k=2 a1=3 a2=3 a3=2，当l=1,r=1或l=2,r=2时满足，那么在i=2时撤销的是0，那么我们选择l=1那这不是两个数吗，其实不是，因为我们放的是sum[l]-l，所以我们在选择他时其实是减去了sum[l]，所以长度还是1是满足条件的

#include<iostream>
#include<cstring>
#include<string>
#include<sstream>
#include<cmath>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<vector> 
#include<set>
#include<unordered_map>
#include<ctime>
#include<cstdlib>
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> PII;
typedef pair<int,pair<int,int> > PIII;
const double eps=1e-7;
const int N=5e5+7 ,M=5e5+7, INF=0x3f3f3f3f,mod=1e9+7;
const long long int llINF=0x3f3f3f3f3f3f3f3f;
inline ll read() {ll x=0,f=1;char c=getchar();while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') {x=(ll)x*10+c-'0';c=getchar();} return x*f;}
inline void write(ll x) {if(x < 0) {putchar('-'); x = -x;}if(x >= 10) write(x / 10);putchar(x % 10 + '0');}
inline void write(ll x,char ch) {write(x);putchar(ch);}
void stin() {freopen("in_put.txt","r",stdin);freopen("my_out_put.txt","w",stdout);}
bool cmp0(int a,int b) {return a>b;}
template<typename T> T gcd(T a,T b) {return b==0?a:gcd(b,a%b);}
template<typename T> T lcm(T a,T b) {return a*b/gcd(a,b);}
void hack() {printf("\n----------------------------------\n");}

int T,hackT;
int n,m,k;
int w[N];
ll sum[N];

void solve() {
	n=read(),k=read();
	
	for(int i=1;i<=n;i++) w[i]=read();
	for(int i=1;i<=n;i++) sum[i]=sum[i-1]+w[i];
	
	ll ans=0;
	map<int,int> st;
	st[0]=1;
	for(int i=1;i<=n;i++) {
		int x=i-k;
		if(x>=0) st[(sum[x]-x+k)%k]--;
		ans+=st[(sum[i]-i+k)%k];
		st[(sum[i]-i+k)%k]++;
	}
	
	write(ans,'\n');
}   

int main() {
    // init();
    // stin();

    // scanf("%d",&T);
    T=1; 
    while(T--) hackT++,solve();
    
    return 0;       
}