WA on #3求助
查看原帖
WA on #3求助
1405239
lbh123楼主2025/1/20 11:18

rt

#include<bits/stdc++.h>
using namespace std;
long long n,k,ans=0,a[100010],sum[100010],fl[100010],sc[1000010],c[100010];
int main(){
	//freopen("ce.out","r",stdin);
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	for(int i=1;i<=n;i++){
		sum[i]=sum[i-1]+a[i];
		if(sum[i]<k){
			fl[i]=0;
			c[i]=k-sum[i];
		}else if(sum[i]%k==0){
			fl[i]=1;
			c[i]=0;
		}else{
			fl[i]=0;
			c[i]=((sum[i]/k)+1)*k-sum[i];
		}
	}
	int t=0,f=0;
	if(fl[1]){
		t++;
		ans++;
	}else{
		f++;
	}
	sc[c[1]]++;
	for(int i=2;i<=n;i++){
		if(fl[i]){
			t++;
			ans+=t;
		}else{
			f++;
			ans+=sc[c[i]];
			sc[c[i]]++;
		}
	}
	cout<<ans;
	return 0;
}
2025/1/20 11:18
加载中...