【动态规划】创意吃鱼法

【动态规划】创意吃鱼法

题目

题目背景

感谢@throusea 贡献的两组数据

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

分析

题目意思：如果某个正方形子矩阵的某条对角线上都有1，且此正方形子矩阵的其他地方无1，则最长为
这个对角线的长度。

先从简单来看，我们先看从左上往右下斜 这条线，
10 ： 如果右下角的位置是1，并且他的 左上是1，正上是0，左边是0，那么他就累计成2，
01

100 ： 如果右下角的位置是1，并且他的左上一条线2个都是1，正上和左边一条线2个都是0，
010 那么他累计成3；
001

那可以得出， 如果一个点是1，他的值等于， min（左上，正上，左边 一条线的值） +1；
这里一条线的值指的是：（正上，左边）从这个点往前延有几个连续的0，就是值的多少。

如果还不懂，再举例：
100 ： 先看中间那个1的位置，他是1，并且他的 左上是1，正上是连续一个0，左边是连续一个0
011 ，那么他就累计成2
001
图变成：
100 ： 看右下那个1的位置，他是1，他的左上是2，正上是连续0个0，左边是连续2个0，
021 min（左上，正上，左边）+1。 可得 他的值：0+1 ==1；
001

这下应该很明白了吧， 右上往左下斜 也是一样的道理。
类似的题还有 最大正方形等等。。

代码

#include<iostream>
#include<cstring>
using namespace std;

const int maxn = 2600;
int n,m;
int a[maxn][maxn];			//读入数据
int b1[maxn][maxn],b2[maxn][maxn];  //存放正上，左边连续的0的情况
int dp[maxn][maxn];

int main(){
	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]==0){
				b1[i][j] = b1[i-1][j]+1;
				b2[i][j] = b2[i][j-1]+1;
			}
			if(a[i][j]==1){
				dp[i][j] = min(dp[i-1][j-1],min(b1[i-1][j],b2[i][j-1]))+1;
				ans =max(ans,dp[i][j]);
			}
		}
	}
	//右上--左下
	memset(b1,0,sizeof(b1)); 
	memset(b2,0,sizeof(b2));
	memset(dp,0,sizeof(dp));
	for(int i=1;i<=n;i++){
		for(int j=m;j>=1;j--){
			if(a[i][j]==0){
				b1[i][j] = b1[i-1][j]+1;
				b2[i][j] = b2[i][j+1]+1;
			}
			if(a[i][j]==1){
				dp[i][j] = min(dp[i-1][j+1],min(b1[i-1][j],b2[i][j+1]))+1;
				ans =max(ans,dp[i][j]);
			}
		}
	} 

	 cout<<ans;
	return 0;
}