用DFS求连通块

例题:UVa 572 (Oil Deposits)
输入一个m行n列的字符矩阵,统计字符'@'组成多少连通块(横,竖或对角线方向)。

****@
*@@*@
*@**@
@@@*@
@@**@

图中有两个连通块。
一般用DFS寻找连通块(DFS好写)。从每个'@'格子出发,递归遍历周围的'@'格子。每次访问一个格子时就给他写上一个“连通分量编号”(下图代码中的idx数组)。这样就可以在访问之前检查它是否有了编号,从而避免同一个格子访问多次。
代码如下:(八连通种子填充算法)

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

const int maxn = 1e2 + 10;

char pic[maxn][maxn];
int m, n, idx[maxn][maxn];

void dfs(int r, int c, int id){
	if(r < 0 || r >= m || c < 0 || c >= n ) return;//出界的格子
	if(idx[r][c] > 0 || pic[r][c] != '@')	return ;//不是'@'或是已经访问过的格子
	idx[r][c] = id;//连通分量编号 
	for(int dr = -1; dr <= 1; dr++)
		for(int dc = -1; dc <= 1; dc ++)
			if(dr != 0 || dc != 0 ) dfs(r+dr, c+dc, id); 
} 

int main(){
	while(scanf("%d %d", &m, &n) == 2 && m && n){
		for(int i = 0; i < m; ++i)	scanf(" %s", pic[i]);
		memset(idx, 0, sizeof idx);
		int cnt = 0;
		for(int i = 0; i < m; ++i)
			for(int j = 0; j < n; ++j)
				if(idx[i][j] == 0 && pic[i][j] == '@')	dfs(i, j, ++cnt);
		printf("%d\n", cnt);
	}
	return 0;
}

猜你喜欢

转载自www.cnblogs.com/wanshe-li/p/12970436.html
今日推荐