题目描述
回到家中的猫猫把三桶鱼全部转移到了她那长方形大池子中,然后开始思考:到底要以何种方法吃鱼呢(猫猫就是这么可爱,吃鱼也要想好吃法 ^_*)。她发现,把大池子视为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
思想:这道题需要预处理下,方便下面计算。前缀和用来保存这个点横向或纵向向前的0的最长个数(即边长)。
状态方程为:dp[ i ][ j ] = min( 上一个状态,横向边长,纵向边长 )。(取最小值是为了围成一个全面符合题意的正方形)。
然后,定义一个ans,每次求dp时更新下就行。
当然,我的写法属于刚好能过的,其实预处理可以和dp写一起的,那样就能少写好多代码。。。
还有输入优化可以节省好多时间,,,以后再补吧。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#define MAX 2505
using namespace std;
int n,m;
int mp[MAX][MAX],dp[MAX][MAX],s1[MAX][MAX],s2[MAX][MAX];
int main(){
int i,j,ans=0;
scanf("%d%d",&n,&m);
memset(dp,0,sizeof(dp));
memset(s1,0,sizeof(s1));
memset(s2,0,sizeof(s2));
//读入数据
for(i=1;i<=n;i++){
for(j=1;j<=m;j++){
scanf("%d",&mp[i][j]);
}
}
//前缀和预处理,统计横向(s1),纵向(s2)该点前最多有几个连续的0
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
if(!mp[i][j])
s1[i][j]=s1[i][j-1]+1;
for(j=1;j<=m;j++)
for(i=1;i<=n;i++)
if(!mp[i][j])
s2[i][j]=s2[i-1][j]+1;
//从左上到右下
for(i=1;i<=n;i++){
for(j=1;j<=m;j++){
if(mp[i][j]){
dp[i][j]=min(min(s1[i][j-1],s2[i-1][j]),dp[i-1][j-1])+1;
ans=max(ans,dp[i][j]);
}
}
}
//前缀和预处理
memset(dp,0,sizeof(dp));
memset(s1,0,sizeof(s1));
for(i=1;i<=n;i++)
for(j=m;j>0;--j)
if(mp[i][j]==0)
s1[i][j]=s1[i][j+1]+1;
//从右上到左下
for(i=1;i<=n;i++){
for(j=m;j>0;--j){
if(mp[i][j]){
dp[i][j]=min(min(s1[i][j+1],s2[i-1][j]),dp[i-1][j+1])+1;
ans=max(ans,dp[i][j]);
}
}
}
printf("%d\n",ans);
return 0;
}