P3941 入阵曲

题目传送门

• memset常数太大，外层16万，不能用
• 要不就在用之前把要用的初始化；
• 要不就每次用完之后把用的置为0，复杂度一样。
• 每次枚举他之前的和他一样的有多少。这样不会有重复

// luogu-judger-enable-o2
#include<bits/stdc++.h>
using namespace std;
const int maxn1=400*1000200;
long long sum[402][402];
int a[401][401];
long long ask(int x1,int y1,int x2,int y2) {
	return sum[x2][y2]-sum[x1-1][y2]-sum[x2][y1-1]+sum[x1-1][y1-1];
}
int need[401],num[401],next[401];
int vis[1000002];
long long ans=0;
int main() {
	int n,m,k;
	cin>>n>>m>>k;
	for(int i=1; i<=n; i++)
		for(int j=1; j<=m; j++) {
			scanf("%d",&a[i][j]);
			sum[i][j]=sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1]+a[i][j];
		}
	for(int l=1; l<=n; l++)//hang
		for(int r=l; r<=n; r++) {
			int now=0,maxx=0;
			//memset(vis,0,sizeof(vis));
			for(int i=1; i<=m; i++) {//lie
				num[i]=(ask(l,i,r,i)%k+num[i-1])%k;
				//	cout<<"num="<<num[i]<<endl;
				vis[num[i]]=0;

			}
			for(int i=1; i<=m; i++) {
				if(num[i]==0)ans+=1;
				ans+=vis[num[i]];//每个数只计算他前面
				vis[num[i]]++;
			}

		}
	cout<<ans;
	return 0;
}