【题目描述】
【Description】
你的老家在河北农村。过年时,你回老家去拜年。你家有一片NM农田,将其看成一个NM的方格矩阵,有些方格是一片水域。你的农村伯伯听说你是学计算机的,给你出了一道题: 他问你:这片农田总共包含了多少个不存在水域的正方形农田。
两个正方形农田不同必须至少包含下面的两个条件中的一条:
边长不相等
左上角的方格不是同一方格
【Input】
输入数据第一行为两个由空格分开的正整数N、M(1<=m 第2行到第N+1行每行有M个数字(0或1),描述了这一片农田。0表示这个方格为水域,否则为农田(注意:数字之间没有空格,而且每行不会出现空格)
【Output】
满足条件的正方形农田个数。
【Sample Input】
3 3
1 1 0
1 1 0
0 0 0
【Sample Output】
5
【解题思路】
这道题我们如果直接求农田的个数会发现问题并不容易解决。
但是,我们可以间接求:定义
d
i
,
j
d_{i,j}
di,j 为前
i
i
i 行
j
j
j 列的正方形最大边长,这样问题就简单了很多。
当我们推动态转移方程时,会发现如果
d
i
,
j
d_{i,j}
di,j 能使最大边长增加的话,那么它
d
i
−
1
,
j
−
1
d_{i-1,j-1}
di−1,j−1 (左上方),
d
i
,
j
−
1
d_{i,j-1}
di,j−1 (左方),
d
i
,
j
−
1
d_{i,j-1}
di,j−1 (上方)三个地方一定都是农田的,则当前位置的最大边长其实就是
d
i
−
1
,
j
−
1
d_{i-1,j-1}
di−1,j−1 ,
d
i
,
j
−
1
d_{i,j-1}
di,j−1,
d
i
,
j
−
1
d_{i,j-1}
di,j−1 中的最小值,如果当前位置不是农田的话则肯定不可能使正方形的最大边长增加。
即
d
i
,
j
=
{
d
i
,
j
=
m
i
n
(
d
i
−
1
,
j
,
d
i
,
j
−
1
,
d
i
−
1
,
j
−
1
)
+
1
m
a
p
i
,
j
=
1
0
m
a
p
i
,
j
=
0
d_{i,j}=\begin{cases} d_{i,j}=min(d_{i-1,j},d_{i,j-1},d_{i-1,j-1})+1&map_{i,j}=1\\ 0&map_{i,j}=0 \end{cases}
di,j={di,j=min(di−1,j,di,j−1,di−1,j−1)+10mapi,j=1mapi,j=0
算出后,我们再来考虑求出问题所求。我们算出正方形农田的最大边长数后把他们直接累加,发现就是问题所求了。
Why?
来证明一下:如果当前位置的最大边长
n
n
n 则 边长为
n
,
n
−
1
,
n
−
2
,
n
−
3
,
.
.
.
,
1
n,n-1,n-2,n-3,...,1
n,n−1,n−2,n−3,...,1 的正方形农田都必定存在,也就是有
n
n
n 块正方形农田,把每个位置累加即为问题所求。
a
n
s
=
∑
i
=
1
n
∑
j
=
1
m
d
i
,
j
ans=\sum_{i=1}^n\sum_{j=1}^md_{i,j}
ans=i=1∑nj=1∑mdi,j
另外需要注意的是
(注意:数字之间没有空格,而且每行不会出现空格)
【CODE】
#include<iostream>
using namespace std;
int n,m;
char a[1011][1011];//注意输入的时候两个数之间没有空格
int d[1011][1011];
int num;
int main()
{
cin>>n>>m;
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
cin>>a[i][j];
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
{
if (a[i][j]=='0')
d[i][j]=0;
else d[i][j]=min(d[i-1][j],min(d[i][j-1],d[i-1][j-1]))+1;//dp求正方形的最大边长
num+=d[i][j];//直接累加
}
cout<<num;
return 0;
}
【PPT里的做法】
(1)设F[I,J]表示以方格(I,J)为右下角,可以得到的最大无水正方形边长,那么显然如果(I,J)是水域,F[I,J]=0;否则F[I,J]=Min{F[I-1,J],F[I,J-1],F[I-1,J-1]}+1
(2)求出了F数组的值后,我们可以用F1[I]表示F数组中,值为I的个数。
(3)显然,假定边长为I的正方形数目为Sum[I],那么有Sum[I]=Sum[I+1]+F1[I]
(4)最后只要算出Sum数组各个值的和为问题的解。
————PPT
【解释】
求最大边长和我的做法一样,但是在求农田个数时出现了偏差。
PPT里的做法是先统计出最大边长为
n
n
n 的有多少个,计算农田个数时,则是将边长为
n
n
n 的正方形分别算出来累加输出。
证明(本蒟蒻可能想的不对):
我们重点来看一下这句话
Sum[I]=Sum[I+1]+F1[I]
我们可以把它理解为边长比
n
n
n 大
1
1
1 的正方形农田也必定存在一个不重复的正方形,加上边长为
n
n
n 的正方型数就是边长为
n
n
n 的正方形总数。
a
n
s
=
∑
i
=
1
n
s
u
m
i
ans=\sum_{i=1}^{n}sum_i
ans=i=1∑nsumi
【CODE】
#include<iostream>
using namespace std;
int n,m;
char a[1011][1011];
int d[1011][1011];
int f[1011];
int sum[1011],num;
int main()
{
cin>>n>>m;
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
cin>>a[i][j];
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
{
if (a[i][j]=='0')
d[i][j]=0;
else d[i][j]=min(d[i-1][j],min(d[i][j-1],d[i-1][j-1]))+1;
f[d[i][j]]++;
}
for (int i=n;i>0;i--)
{
sum[i]=sum[i+1]+f[i];
num+=sum[i];
}
cout<<num;
return 0;
}
【总结】
个人推荐方法一又好理解时间和空间复杂度又比第二种优。