【JZOJ 4817】【NOIP2016提高组 五校联考4】square

Description

Analysis

我们可以想到一个DP，f[i][j]表示(i,j)为右下角的最大正方形边长。
则$f[i][j]=Min(f[i-1][j-1],f[i-1][j],f[i][j-1])+1$
那么可以二分答案，在合法范围内找是否有f[x][y]>=mid。
时间复杂度$O(Tn^2logn)$，超时。
注意到f是静态的，有什么东西可以静态查询区间最大值？
没错，就是RMQ，不过是二维的。
其实二维rmq也差不多，设rmq[k][l][i][j]表示以(i,j)为右下角，向上2^i个格子，向左2^j个格子的矩形内f的最大值。那么预处理rmq数组就是$O(n^2log^2n)$
对于二分出来的数，有了rmq数组我们就可以O(1)判断，所以时间复杂度为$O(Tlogn)$

Code

#include<cstdio>
#include<cmath>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=1005;
int n,m,_2[11],f[N][N],rmq[11][11][N][N];
bool map[N][N];
char ch;
int read()
{
    int t=0,p=1;
    for(ch=getchar();ch<'0' || ch>'9';ch=getchar())
        if(ch=='-') p=-1;
    for(;'0'<=ch && ch<='9';ch=getchar()) t=t*10+ch-'0';
    return t*p;
}
int query(int x1,int y1,int x2,int y2)
{
    if(x1>x2 || y1>y2) return 0;
    int l1=log2(x2-x1+1),l2=log2(y2-y1+1);
    int ans=max(rmq[l1][l2][x2][y2],rmq[l1][l2][x2][y1+_2[l2]-1]);
    ans=max(ans,max(rmq[l1][l2][x1+_2[l1]-1][y2],rmq[l1][l2][x1+_2[l1]-1][y1+_2[l2]-1]));
    return ans;
}
bool check(int x1,int y1,int x2,int y2,int l)
{
    return query(x1+l-1,y1+l-1,x2,y2)>=l;
}
int main()
{
    freopen("square.in","r",stdin);
    freopen("square.out","w",stdout);
    _2[0]=1;
    fo(i,1,10) _2[i]=_2[i-1]*2;
    int _,x1,y1,x2,y2;
    n=read(),m=read();
    fo(i,1,n)
        fo(j,1,m)
        {
            map[i][j]=read();
            if(map[i][j]) f[i][j]=min(min(f[i-1][j],f[i][j-1]),f[i-1][j-1])+1;
            rmq[0][0][i][j]=f[i][j];
        }
    fo(k,1,log2(n))
        fo(i,1,n)
            fo(j,1,m)
            {
                rmq[0][k][i][j]=max(rmq[0][k-1][i][j],rmq[0][k-1][i][j-_2[k-1]]);
                rmq[k][0][i][j]=max(rmq[k-1][0][i][j],rmq[k-1][0][i-_2[k-1]][j]);
            }
    fo(k,1,log2(n))
        fo(l,1,log2(m))
            fo(i,_2[k],n)
                fo(j,_2[l],m)
                    rmq[k][l][i][j]=max(rmq[k][l-1][i][j],rmq[k][l-1][i][j-_2[l-1]]);
    _=read();
    while(_--)
    {
        x1=read(),y1=read(),x2=read(),y2=read();
        int l=0,r=min(y2-y1+1,x2-x1+1);
        while(l+1<r)
        {
            int mid=(l+r)>>1;
            if(check(x1,y1,x2,y2,mid)) l=mid;
            else r=mid-1;
        }
        while(check(x1,y1,x2,y2,l+1)) l++;
        printf("%d\n",l);
    }
    return 0;
}