创新吃鱼法

http://codevs.cn/problem/1773/
题目描述 Description
回到家中的猫猫把三桶鱼全部转移到了她那长方形大池子中，然后开始思考：到底要以何种方法吃鱼呢（猫猫就是这么可爱，吃鱼也要想好吃法 ^_*）。她发现，把大池子视为01矩阵（0表示对应位置无鱼，1表示对应位置有鱼）有助于决定吃鱼策略。

在代表池子的01矩阵中，有很多的正方形子矩阵，如果某个正方形子矩阵的某条对角线上都有鱼，且此正方形子矩阵的其他地方无鱼，猫猫就可以从这个正方形子矩阵“对角线的一端”下口，只一吸，就能把对角线上的那一队鲜鱼吸入口中。

　　猫猫是个贪婪的家伙，所以她想一口吃掉尽量多的鱼。请你帮猫猫计算一下，她一口下去，最多可以吃掉多少条鱼？

输入描述 Input Description
输入数据：

　　有多组输入数据，每组数据：

　　第一行有两个整数n和m（n,m≥1），描述池塘规模。接下来的n行，每行有m个数字（非“0”即“1”）。每两个数字之间用空格隔开。

　　对于30%的数据，有n,m≤100

　　对于60%的数据，有n,m≤1000

　　对于100%的数据，有n,m≤2500

输出描述 Output Description
输出数据：

　　只有一个整数——猫猫一口下去可以吃掉的鱼的数量，占一行，行末有回车。

样例输入 Sample Input
输入样例：

4 6

0 1 0 1 0 0

0 0 1 0 1 0

1 1 0 0 0 1

0 1 1 0 1 0

样例输出 Sample Output
输出样例：

3

#include <iostream>
#include <cstring>
using namespace std;
int a[2502][2502],l[2502][2502],u[2502][2502],f[2502][2502];
int main()
{
    std::ios::sync_with_stdio(false);
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        a[i][0]=1;
        a[i][m+1]=1;
    }
    for(int i=1;i<=m;i++)
    {
        a[0][i]=1;
    }//封起来，防止越界
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            cin>>a[i][j];
            if(a[i][j-1]==0)l[i][j]=l[i][j-1]+1;//列
            else l[i][j]=0;
            if(a[i-1][j]==0)u[i][j]=u[i-1][j]+1;//行
            else u[i][j]=0;
        }
    }
    int ans=0;

    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            if(a[i][j]==1)
            {
                f[i][j]=min(min(f[i-1][j-1],l[i][j]),u[i][j])+1;
            }
            else f[i][j]=0;
            ans=max(ans,f[i][j]);
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=m;j>=1;j--)
        {
            if(a[i][j+1]==0)
            {
                l[i][j]=l[i][j+1]+1;//换一个方向，斜向左
            }
            else l[i][j]=0;
        }
    }
    memset(f,0,sizeof(f));
    for(int i=1;i<=n;i++)
    {
        for(int j=m;j>=1;j--)
        {
            if(a[i][j]==1)f[i][j]=min(min(f[i-1][j+1],l[i][j]),u[i][j])+1;
            else f[i][j]=0;
            ans=max(ans,f[i][j]);
        }
    }
    cout<<ans<<endl;
    return 0;
}