洛谷 P2363 马农

题目描述

分别枚举两个矩阵？那样n^6太要命了。

可以枚举两个矩形的交点
将交点看成原点，可以将整个区域分成四个象限，1与3对应，2与4对应
再枚举相对应的象限计算可以获得的利益，用hash判重

可枚举不同的象限时还要把hash清零，n^2次的memset就超时了。。。

那怎么继续优化呢?

可以用一个栈记录hash里增加的值，只把这些值清零就好了。

#include<complex>
#include<cstdio>
using namespace std;
const int N=,M=1e6+;
int n,ans,top;
int st[N*N],sum[N][N],Hash[M<<];
int main()
{
    scanf("%d",&n);
    int a;
    for(int i=;i<=n;i++)
        for(int j=;j<=n;j++)
        {
            scanf("%d",&a);
            sum[i][j]=sum[i-][j]+sum[i][j-]-sum[i-][j-]+a;
        }
    int tmp;
    for(int i=;i<=n-;i++)
        for(int j=;j<=n-;j++)
        {
            for(int k=;k<=i;k++)
                for(int l=;l<=j;l++)
                {
                    tmp=sum[i][j]-sum[k-][j]-sum[i][l-]+sum[k-][l-]+M;//可能会有负数,所以都加M变为正数
                    st[++top]=tmp;
                    Hash[tmp]++;
                }
            for(int k=i+;k<=n;k++)
                for(int l=j+;l<=n;l++)
                {
                    tmp=sum[i][j]-sum[k][j]-sum[i][l]+sum[k][l]+M;
                    ans+=Hash[tmp];
                }
            while(top)Hash[st[top--]]=;
            for(int k=i+;k<=n;k++)
                for(int l=;l<=j;l++)
                {
                    tmp=sum[i][l-]-sum[k][l-]-sum[i][j]+sum[k][j]+M;
                    st[++top]=tmp;
                    Hash[tmp]++;
                }
            for(int k=;k<=i;k++)
                for(int l=j+;l<=n;l++)
                {
                    tmp=sum[i][l]-sum[k-][l]-sum[i][j]+sum[k-][j]+M;
                    ans+=Hash[tmp];
                }
            while(top)Hash[st[top--]]=;
        }
    printf("%d\n",ans);
    return ;
}