luogu P1736 创意吃鱼法

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<string>
#include<map>
typedef long long ll;
using namespace std;
int n,m,ans;
int a[2509][2509],f[2509][2509],s1[2509][2509],s2[2509][2509];//s1为横向，s2为纵向

//s1[i][j]表示（i，j）最多向左(右)延伸多少个格子，使这些格子中的数都是0（不包括（i，j）） 
//s2[i][j]表示（i，j）最多向上(左)延伸多少个格子，使这些格子中的数都是0（不包括（i，j））  
//f[i][j]表以（i，j）为右下（左下）角的最大对角线长度 
int main()
{
    cin>>n>>m;
    //第一遍左上——右下
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
        {
            scanf("%d",&a[i][j]);
            //如果是0 
            if(!a[i][j])
            {
                //向左 
                s1[i][j]=s1[i][j-1]+1;
                //向上 
                s2[i][j]=s2[i-1][j]+1;
            }
            //如果是1 
            if(a[i][j])
                // f[i-1][j-1]和 (s1[i][j-1],s2[i-1][j])同时取小
                //为了防止有1出现
                f[i][j]=min(f[i-1][j-1],min(s1[i][j-1],s2[i-1][j]))+1;
            ans=max(ans,f[i][j]);
        }
    //第二遍右上——左下
    memset(f,0,sizeof(f));
    memset(s1,0,sizeof(s1));//数组置0
    memset(s2,0,sizeof(s2));
    for(int i=1; i<=n; i++)
        for(int j=m; j>=1; j--)
        {
            if(!a[i][j])
            {
                s1[i][j]=s1[i][j+1]+1;
                s2[i][j]=s2[i-1][j]+1;
            }
            if(a[i][j])
                f[i][j]=min(f[i-1][j+1],min(s1[i][j+1],s2[i-1][j]))+1;
            ans=max(ans,f[i][j]);
        }
    cout<<ans<<endl;
    return 0;
}