连通区域求解 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 在下一篇分享中增加最大连通区域的求解