50分蒟蒻在线求助

P3941:50pts

#include<bits/stdc++.h>
using namespace std;
long long sum[405][405],cnt[1000005],mod[1000005],ans;
int n,m,k,a[405][405];
int main(){
	scanf("%d%d%d",&n,&m,&k);
	for(register int i=1;i<=n;++i)
		for(register int j=1;j<=m;++j)
			scanf("%d",&a[i][j]),sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j];
	for(register int i=0;i<n;++i)
		for(register int j=i+1;j<=n;++j){
			cnt[0]=1;
			for(register int l=1;l<=m;++l){
				mod[l]=(sum[j][l]-sum[i][l]+k)%k;
				ans+=cnt[mod[l]];
				++cnt[mod[l]];
			}
			memset(cnt,0,sizeof(cnt));
		}
	printf("%lld",ans);
    return 0;
}