uva352 The Seasonal War-python

最新推荐文章于 2020-03-09 21:40:41 发布

原创最新推荐文章于 2020-03-09 21:40:41 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #uva

算法专栏收录该内容

40 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）算法来计算二维矩阵中连通块数量的方法。通过遍历每个元素并标记已访问区域，有效地统计了矩阵中所有独立的连通区域。代码示例清晰地展示了DFS的应用过程。

计算连通块，dfs

MAXN = 26
dirs = [[1, 0], [0, 1], [-1, 0], [0, -1], [1, 1], [1, -1], [-1, 1], [-1, -1]]
cnt = 0


def dfs(x, y, m, n):
    flag[x][y] = 1
    for d in dirs:
        nx = x + d[0]
        ny = y + d[1]
        if 0 <= nx < m and 0 <= ny < n and pic[nx][ny] == '1' and flag[nx][ny] == 0:
            dfs(nx, ny, m, n)
while True:
    nn = raw_input()
    if not nn.isdigit():
        break
    n = int(nn)
    flag = [[0 for col in range(MAXN)] for row in range(MAXN)]
    pic = []
    ans = 0
    cnt += 1
    for i in range(n):
        pic.append(raw_input())

    for i in range(n):
        for j in range(n):
            if pic[i][j] == '1' and flag[i][j] == 0:
                dfs(i, j, n, n)
                ans += 1
    print("Image number "+str(cnt)+" contains "+str(ans)+" war eagles.")