这个题和最大正方形的题目真的很像,然而最大正方形实质上就是二维前缀和乱搞(虽然第一眼我没想到)。这个题可以不用预处理,维护sl和sc两个数组,好吧一开始前一天的半夜看错了题意。根据题意,就是求最长的符合要求的对角线,对角线有两条,所以一次从左向右,一次从右到左,根据动态规划的逻辑,以从左到右为例,dp[i][j]应该来自dp[i-1][j-1],以及sl[i][j-1]和sc[i-1][j],取他们中的最小值加a[i][j]就是dp[i][j]。然后看代码吧。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<bitset>
#include<vector>
#include<algorithm>
#include<set>
#include<map>
using namespace std;
const int maxn=2507;
int dp[maxn][maxn],sl[maxn][maxn],sc[maxn][maxn],a[maxn][maxn];
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
memset(dp,0,sizeof(dp));
memset(sl,0,sizeof(sl));
memset(sc,0,sizeof(sc));
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
scanf("%d",&a[i][j]);
int ans=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
if(!a[i][j])
{
sl[i][j]=sl[i][j-1]+1;
sc[i][j]=sc[i-1][j]+1;
}
if(a[i][j])
{
dp[i][j]=min(dp[i-1][j-1],min(sl[i][j-1],sc[i-1][j]))+1;
ans=max(ans,dp[i][j]);
}
}
}
memset(dp,0,sizeof(dp));
memset(sl,0,sizeof(sl));
memset(sc,0,sizeof(sc));
for(int i=1;i<=n;i++)
for(int j=m;j>=1;j--)
{
if(!a[i][j])
{
sl[i][j]=sl[i][j+1]+1;
sc[i][j]=sc[i-1][j]+1;
}
if(a[i][j])
{
dp[i][j]=min(dp[i-1][j+1],min(sl[i][j+1],sc[i-1][j]))+1;
ans=max(ans,dp[i][j]);
}
}
printf("%d\n",ans);
}
}