#include<bits/stdc++.h>
using namespace std;
// dfs 二维空间统计四连通区域个数,并返回每个连通区域中的节点个数
int n = 3, m = 3, ans;
int vis[100], mp[100];
int dfs (int x, int y) { // 这个函数必须要传入行号和列号,不能传入降维至一维的编号
if (x < 0 || x >= n || y < 0 || y >= m) return 0;
if (vis[x*m+y]) return 0; // 这个是表示深搜过的点不去搜索了
if (!mp[x*m+y]) return 0;
vis[x*m+y] = 1;
mp[x*m+y] = 0; // 遍历过的地方要去掉标记,要不然主函数中后面的循环还以该点为起点深搜。
return 1 + dfs (x - 1, y) + dfs (x + 1, y) + dfs (x, y - 1) + dfs (x, y + 1);
}
int main()
{
mp[1*m+1] = mp[0*m+1] = mp[0*m+0] = 1;
mp[2*m+2] = 1;
for (int i = 0;i < n*m;i ++)
if (mp[i])
memset (vis, 0, sizeof vis),
ans ++,
cout << "第" << ans << "个:" << dfs (i/m, i%m) << endl;
return 0;
}
该模板是用于暴力判断一个二维空间中连通区域的个数,以及每个连通区域由多少块组成。
特别强调,dfs函数的参数必须传入行号和列号,不能传入降至一维的表示。
因为当原行号和列号存在非法的情况时,由一维编号计算出的行号和列号未必是原行号和列号,比如:行数为3,列数为4,原行号为1,原列号为-1,那么对应的一维编号为1*4-1 = 3,虽然一维编号合法,但其实原坐标是不合法的。
如果传入参数为一维坐标,该函数不能保证传入的一维坐标对应的二维坐标一定是合法的,因此使用“除以m”和“取模m”的操作得到的二维坐标并不一定是原坐标,而且由一维坐标得到的二维坐标一定是合法的,所以如果参数使用一维坐标表示,那么第一if判断中的一维坐标%m>=m
是永远成立不了的,最终导致错误。
更新于2022.4.7(第十三届蓝桥杯省赛前前夕)
写了个更简洁的版本,但是由于不适用vis数组所以会导致dfs的时候更改初始输入的地图。
#include<bits/stdc++.h>
using namespace std;
const int N = 100;
int n, m, ans, a[N][N];
int dfs (int x, int y) {
if (x < 0 || y < 0 || x >= n || y >= m || !a[x][y]) return 0;
a[x][y] = 0;
return dfs (x-1, y) + dfs (x+1, y) + dfs (x, y-1) + dfs (x, y+1) + 1;
}
int main()
{
cin >> n >> m;
for (int i = 0;i < n;i ++)
for (int j = 0;j < m;j ++)
cin >> a[i][j];
for (int i = 0;i < n;i ++)
for (int j = 0;j < m;j ++)
if (a[i][j])
ans ++,
cout << dfs (i, j) << endl;
cout << "num is " << ans << endl;
return 0;
}
/*
4 4
1 0 1 1
1 1 0 1
1 0 1 0
0 0 0 1
4 4
0 0 0 0
0 1 1 1
0 1 0 1
0 1 1 1
*/