解题思路:对角线可以是从左上角到右下角,也可以是从右上角到左下角,所以答案需要求两次,去最大的。先讨论从左上角到右下角。dp[i][j]为以(i,j)点为右端点的能吃到的最多的鱼,由于需要子矩阵的其他地方没有鱼,所以需要两个辅助数组,分别统计(i,j)这点向上有多少个连续的0,向左有多少个连续的0。所以我们可以得到 dp[i][j]=min(dp[i−1][j−1], up[i−1][j], left[i][j−1])+1
从右上角到左下角也是一样的讨论。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f;
const int maxn=2505;
int n,m,a[maxn][maxn];
int dp[maxn][maxn],up[maxn][maxn],row[maxn][maxn];
int main(){
std::ios::sync_with_stdio(0);
cin>>n>>m;
int ans=0;
for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j){
cin>>a[i][j];
if(a[i][j]){
dp[i][j]=min(dp[i-1][j-1],min(up[i-1][j],row[i][j-1]))+1;
ans=max(dp[i][j],ans);
}
else{
up[i][j]=up[i-1][j]+1;
row[i][j]=row[i][j-1]+1;
}
}
}
memset(dp,0,sizeof(dp));
memset(row,0,sizeof(row));
for(int i=1;i<=n;++i){
for(int j=m;j>=1;--j){
if(a[i][j]){
dp[i][j]=min(dp[i-1][j+1],min(up[i-1][j],row[i][j+1]))+1;
ans=max(dp[i][j],ans);
}
else{
row[i][j]=row[i][j+1]+1;
}
}
}
cout<<ans<<endl;
return 0;
}