算法课上的一道作业。
题目
在观看完战马检阅之后,来自大草原的两兄弟决心成为超级”马农”,专门饲养战马。兄弟两回到草原,将可以养马的区域,分为N*N的单位面积的正方形,并实地进行考察,归纳出了每个单位面积可以养马所获得的收益。接下来就要开始规划他们各自的马场了。首先,两人的马场都必须是矩形区域。同时,为了方便两人互相照应,也为了防止马匹互相走散,规定两个马场的矩形区域相邻,且只有一个交点。最后,互不认输的两人希望两个马场的收益相当,这样才不会影响他们兄弟的感情。现在,兄弟两找到你这位设计师,希望你给他们设计马场,问共有多少种设计方案。
Input
第一行一个整数N,表示整个草原的大小为N*N。接下来N行,每行N个整数A(i,j),表示第i行第j列的单位草地的收成。(注意:收益可能是负数,养马也不是包赚的,马匹也可能出现生病死亡等意外。)N<=50, -1000
Output
输出符合两人要求的草原分配方案数。
Simple Input
3
1 2 3
4 5 6
7 8 9
Simple Output
2
Hint
思路:穷举每个矩形的交点,每个交点会把整个区域划分成四块,如下图
①区域对应④区域,③区域对应②区域。分别穷举每个区域里的矩形,注意矩形的一个顶点一定为交点。然后用哈希记录收益值,找出对应区域相等收益的个数(这里T了一发,因为本人很懒一般不写哈希,直接用map做)。还有一个优化,在代码注释里面有介绍。
下面是代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<map>
#include<algorithm>
#define LL long long
#define M 2500000
using namespace std;
int n,sum[55][55];
int ans,x,tmp,top;
int h[M*2+10],st[55*55];
int main()
{
while(~scanf("%d",&n))
{
memset(sum,0,sizeof(sum));
ans=top=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
scanf("%d",&x);
sum[i][j]=sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1]+x;
}
}
for(int i=1;i<n;i++)
{
for(int j=1;j<n;j++) //这两个for穷举了交点
{
//cout<<sum[i][j]<<' ';
for(int k=1;k<=i;k++)
{
for(int l=1;l<=j;l++) //这两个是穷举区域里面的矩形
{
tmp=sum[i][j]+sum[k-1][l-1]-sum[k-1][j]-sum[i][l-1]+M;
st[top++]=tmp; //用哈希记录 ,用map超时orz
h[tmp]++;
}
}
for(int k=i+1;k<=n;k++)
{
for(int l=j+1;l<=n;l++)
{
tmp=sum[i][j]+sum[k][l]-sum[k][j]-sum[i][l]+M;
ans+=h[tmp];
}
}
while(top) //这里用了一个堆栈记录哈希的个数,直接清空对应的
h[st[--top]]=0; //如果直接清空整个数组,应该是会超时的
for(int k=i+1;k<=n;k++)
{
for(int l=1;l<=j;l++)
{
tmp=sum[i][l-1]+sum[k][j]-sum[i][j]-sum[k][l-1]+M;
st[top++]=tmp;
h[tmp]++;
}
}
for(int k=1;k<=i;k++)
{
for(int l=j+1;l<=n;l++)
{
tmp=sum[i][l]+sum[k-1][j]-sum[k-1][l]-sum[i][j]+M;
ans+=h[tmp];
}
}
while(top)
h[st[--top]]=0;
}
//cout<<endl;
}
printf("%d\n",ans);
}
}