【洛谷P2363】马农
枚举两个矩形,一共需要枚举三个点,那么时间复杂度就是\(O(n^6)\),面对数据范围无法接受。
再重新想一下问题,两个矩形的权值和相同,就会对答案产生一的贡献,所以我们可以用空间换时间,也就是我们先枚举其中一个矩形,然后把所有的情况也就是权值全部放到桶里,然后在枚举另外一个矩形,对于出现相同的权值和时,用乘法原理搞一下就好了。
有一个需要注意的问题就是,对于我们的桶,他是很大的,而且是两个,所以我们memset是很亏的,也是一定超时的。
那么怎么办,直接暴力再枚举一遍把所有的权值在桶里减回去就好了。
code:
#include<iostream>
#include<cstdio>
#include<cstring>
#define int long long
using namespace std;
inline int read(){
int sum=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0';ch=getchar();}
return sum*f;
}
int sum[505][505];
int t[2500070],tt[2500070];
int n,x,ans;
signed main(){
n=read();
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
int x;
x=read();
sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+x;
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
for(int k=1;k<=i;k++){
for(int l=1;l<=j;l++){
int tmp=sum[i][j]-sum[k-1][j]-sum[i][l-1]+sum[k-1][l-1];
t[tmp]++;
}
}
for(int k=i+1;k<=n;k++){
for(int l=j+1;l<=n;l++){
int tmp=sum[k][l]-sum[i][l]-sum[k][j]+sum[i][j];
if(t[tmp]){
ans-=t[tmp]*tt[tmp];
tt[tmp]++;
ans+=t[tmp]*tt[tmp];
}
}
}
for(int k=1;k<=i;k++){
for(int l=1;l<=j;l++){
int tmp=sum[i][j]-sum[k-1][j]-sum[i][l-1]+sum[k-1][l-1];
t[tmp]--;
}
}
for(int k=i+1;k<=n;k++){
for(int l=j+1;l<=n;l++){
int tmp=sum[k][l]-sum[i][l]-sum[k][j]+sum[i][j];
tt[tmp]--;
}
}
for(int k=1;k<=i;k++){
for(int l=j+1;l<=n;l++){
int tmp=sum[i][l]-sum[i][j]-sum[k-1][l]+sum[k-1][j];
t[tmp]++;
}
}
for(int k=i+1;k<=n;k++){
for(int l=1;l<=j;l++){
int tmp=sum[k][j]-sum[k][l-1]-sum[i][j]+sum[i][l-1];
if(t[tmp]){
ans-=t[tmp]*tt[tmp];
tt[tmp]++;
ans+=t[tmp]*tt[tmp];
}
}
}
for(int k=1;k<=i;k++){
for(int l=j+1;l<=n;l++){
int tmp=sum[i][l]-sum[i][j]-sum[k-1][l]+sum[k-1][j];
t[tmp]--;
}
}
for(int k=i+1;k<=n;k++){
for(int l=1;l<=j;l++){
int tmp=sum[k][j]-sum[k][l-1]-sum[i][j]+sum[i][l-1];
tt[tmp]--;
}
}
}
}
printf("%lld\n",ans);
return 0;
}