簡體   English   中英

Python-確定矩陣單元格是一組

[英]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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM