[英]Python - Determine Matrix cells are a group
給定一組元組,它們表示矩陣中的位置。
for example {(1, 2), (1, 3), (2, 2), (0, 3), (0, 4)}
其中元組(r,k)代表行r和列k。 我如何確定它們是否掛在一起?
examples
{(1, 2), (1, 3), (2, 2), (0, 3), (0, 4)} => hangs together
{(1, 2), (1, 4), (2, 2), (0, 3), (0, 4)} => doesnt hang together
我只是做簡單的BFS或DFS,例如:
def connected(cells):
if cells:
cells = cells.copy()
stack = [cells.pop()]
while stack:
i, j = stack.pop()
neighbors = {(i-1, j), (i+1, j), (i, j-1), (i, j+1)} & cells
stack.extend(neighbors)
cells -= neighbors
return not cells
用法/演示:
for cells in ({(1, 2), (1, 3), (2, 2), (0, 3), (0, 4)},
{(1, 2), (1, 4), (2, 2), (0, 3), (0, 4)}):
print(connected(cells))
打印:
True
False
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.