连通区域求解 python

连通区域求解 python

题目：对一个矩阵（列表），求解其所有的联通区域，并输出

思路：用回归的方法先进行标记，然后再进行遍历

'''
1, 0, 0, 0, 1
0, 1, 0, 1, 1
1, 0, 1, 0, 1
1, 0, 1, 0, 0
'''
'用列表代替矩阵操作'
table = [
    [1, 0, 0, 0, 1],
    [0, 1, 0, 1, 1],
    [1, 0, 1, 0, 1],
    [1, 0, 1, 0, 0]
]
'定义三个全局变量'
rows = len(table)
cols = len(table[0])
label_table = []

'生成对应的标记列表'
for i in range(rows):
    col = cols * [0]
    label_table.append(col)

'按行打印'
def show(table):
    for i in range(rows):
        print(table[i])
    print()

'利用回归进行标记'
def dfs(i, j, mask):  
    if i<0 or i>=rows or j<0 or j>=cols or label_table[i][j] != 0 or table[i][j] != 1:
        return 0
    else:
        label_table[i][j] = mask
        dfs(i, j - 1, mask)
        dfs(i, j+1, mask)
        dfs(i-1, j, mask)
        dfs(i+1, j, mask)
        return 1

if __name__ == "__main__":
    print("label table")
    mask = 1
    for i in range(rows):
        for j in range(cols):
            if table[i][j] == 1 and label_table[i][j] == 0:
                dfs(i, j, mask)
                mask += 1
    show(label_table)

P.S 在下一篇分享中增加最大连通区域的求解