洛谷 P1736 创意吃鱼法

题目描述

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

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

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

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

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

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

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

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

输出格式：

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

输入输出样例
输入样例#1：

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

输出样例#1：

3

说明

右上角的

1 0 0 0 1 0 0 0 1

---

【分析】
依然是高端的 前缀和+dp
对于一个为1的点，左边连续为0的个数，上边连续为0的个数，斜上角的dp[i-1][j-1]取一个min再+1就是答案。
注意还要统计右上角的 dp[i-1][j+1]

---

【代码】

//创意吃鱼法 
#include<iostream>
#include<vector>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define M(a) memset(a,0,sizeof a)
#define of(i,j,k) for(register int i=j;i>=k;i--) 
#define fo(i,j,k) for(register int i=j;i<=k;i++)
using namespace std;
const int mxn=2505;
int n,m,ans;
int resh[mxn][mxn],resl[mxn][mxn],a[mxn][mxn],dp[mxn][mxn];
int main()
{
    scanf("%d%d",&n,&m);
    fo(i,1,n)
      fo(j,1,m)
      {
          scanf("%d",&a[i][j]);
          if(!a[i][j])
            resh[i][j]=resh[i][j-1]+1,
            resl[j][i]=resl[j][i-1]+1;
      }
    fo(i,1,n) of(j,m,1) if(!a[i][j]) resh[i][j]=max(resh[i][j],resh[i][j+1]);
    fo(i,1,n)
      fo(j,1,m)
        if(a[i][j])
        {
            dp[i][j]=1;
            dp[i][j]=min(dp[i-1][j-1],min(resh[i][j-1],resl[j][i-1]))+1;
        }
    fo(i,1,n) fo(j,1,m) ans=max(ans,dp[i][j]);
    M(dp);
    fo(i,1,n)
      fo(j,1,m)
        if(a[i][j])
        {
            dp[i][j]=1;
            dp[i][j]=min(dp[i-1][j+1],min(resh[i][j+1],resl[j][i-1]))+1;
        }
    fo(i,1,n) fo(j,1,m) ans=max(ans,dp[i][j]);
    printf("%d\n",ans);
    return 0;    
}