P1736 创意吃鱼法（dp+前缀和优化）

题目描述

回到家中的猫猫把三桶鱼全部转移到了她那长方形大池子中，然后开始思考：到底要以何种方法吃鱼呢（猫猫就是这么可爱，吃鱼也要想好吃法 ^_*）。她发现，把大池子视为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;
}