用DFS求连通块(种子填充)

【问题】

输入一个m行n列的字符矩阵,统计字符“@”组成多少个八连块。如果两个字符“@”所在
的格子相邻(横、竖或者对角线方向),就说它们属于同一个八连块。例如,图6-9中有两
个八连块。

  图6-9

【分析】

和前面的二叉树遍历类似,图也有DFS和BFS遍历。由于DFS更容易编写,一般用DFS找
连通块:从每个“@”格子出发,递归遍历它周围的“@”格子。每次访问一个格子时就给它写
上一个“连通分量编号”(即下面代码中的idx数组),这样就可以在访问之前检查它是否已经
有了编号,从而避免同一个格子访问多次

#include<cstdio>
#include<cstring>
const int maxn = 100 + 5;
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; //连通分量编号
    
    //接收主函数传过来的格子
    //从这个格子出发递归向八个方向走
    //走完以后cnt++,表示这是一个连通量
    
    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/remly/p/10417465.html