洛谷 2484 [SDOI2011]打地鼠

【题解】

　　n^6的做法很好想，然而这样复杂度不对。。

　　然后我们可以发现R和C可以分开求，这样复杂度降到了n^4. 使用树状数组可以把复杂度降到n^3logn，可以顺利通过。

　　

#include<cstdio>
#include<algorithm>
#define N 1010
#define rg register
#define lowbit (x&-x)
using namespace std;
int n,m,r,c,ans,a[N][N],b[N],t[N];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
inline void add(int x,int y){
    for(;x<=n;x+=lowbit) t[x]+=y;
}
inline int query(int x){
    int ret=0; for(;x;x-=lowbit) ret+=t[x]; return ret;
}
inline bool check1(int len){
    for(rg int j=1;j<=n;j++){
        for(rg int i=1;i<=m;i++) b[i]=a[j][i];
        for(rg int i=1;i<=m-len+1;i++){
            if(b[i]<0) return 0;
            for(rg int k=i+1;k<=i+len-1;k++) b[k]-=b[i]; b[i]=0;
        }
        for(rg int i=m-len+2;i<=m;i++) if(b[i]!=0) return 0;
//        for(rg int i=1;i<=m;i++) printf("%d ",b[i]); puts(""); 
    }
    return 1;
}
inline bool check2(int len){
    for(rg int j=1;j<=m;j++){
        for(rg int i=1;i<=n;i++) b[i]=a[i][j];
        for(rg int i=1;i<=n-len+1;i++){
            if(b[i]<0) return 0;
            for(rg int k=i+1;k<=i+len-1;k++) b[k]-=b[i]; b[i]=0;
        }
        for(rg int i=n-len+2;i<=n;i++) if(b[i]!=0) return 0;
//        for(rg int i=1;i<=m;i++) printf("%d ",b[i]); puts(""); 
    }
    return 1;
}
int main(){
    n=read(); m=read();
    for(rg int i=1;i<=n;i++)
        for(rg int j=1;j<=m;j++) a[i][j]=read(),ans+=a[i][j];
    for(rg int i=m;i>=1;i--) if(check1(i)){
        c=i; break;
    }
    for(rg int i=n;i>=1;i--) if(check2(i)){
        r=i; break;
    }
    printf("%d
",ans/r/c);
    return 0;
}