python - 识别矩阵中最大的连通分量

Question

我有一个带有 1 和 0 的 python numpy 矩阵，我需要确定矩阵中 1 的最大“集合”：http: //imgur.com/4JPZufS

矩阵最多可以有 960.000 个元素，所以我想避免暴力解决方案。

解决这个问题的最聪明的方法是什么？

Answer 1

您可以使用一种名为disjoint-set的数据结构（这里是一个 python 实现）。这种数据结构就是为这种任务而设计的。

如果当前元素为 1，则迭代行，检查是否有任何已遍历的邻居为 1。如果是，则将此元素添加到其集合中。如果这些集合有超过 1 个联合。如果没有邻居是 1，则创建一个新集合。最后输出最大的集合。

这将按如下方式工作：

def MakeSet(x):
  x.parent = x
  x.rank   = 0
  x.size = 1

def Union(x, y):
  xRoot = Find(x)
  yRoot = Find(y)
  if xRoot.rank > yRoot.rank:
    yRoot.parent = xRoot
  elif xRoot.rank < yRoot.rank:
    xRoot.parent = yRoot
  elif xRoot != yRoot: # Unless x and y are already in same set, merge them
    yRoot.parent = xRoot
    xRoot.rank = xRoot.rank + 1
  x.size += y.size
  y.size = x.size

def Find(x):
  if x.parent == x:
    return x
  else:
    x.parent = Find(x.parent)
    return x.parent

""""""""""""""""""""""""""""""""""""""""""

class Node:
  def __init__ (self, label):
    self.label = label
  def __str__(self):
    return self.label

rows = [[1, 0, 0], [1, 1, 0], [1, 0, 0]]
setDict = {}
for i, row in enumerate(rows):
  for j, val in enumerate(row):
    if row[j] == 0:
      continue
    node = Node((i, j))
    MakeSet(node)
    if i > 0:
      if rows[i-1][j] == 1:
        disjointSet = setDict[(i-1, j)]
        Union(disjointSet, node)
    if j > 0:
      if row[j-1] == 1:
      disjointSet = setDict[(i, j-1)]
      Union(disjointSet, node)
    setDict[(i, j)] = node
print max([l.size for l in setDict.values()])

>> 4

这是一个完整的工作示例，其中包含从上面的链接中获取的不相交集的代码。

Answer 2

我认为计数将在提供的答案中取消。例如，如果行更改为rows = [[1, 0, 0], [1, 1, 1], [1, 0, 0]]仍然得到 4，尽管它应该是 5。将联合更改为

def Union(x, y):
  xRoot = Find(x)
  yRoot = Find(y)
  if xRoot.rank > yRoot.rank:
    yRoot.parent = xRoot
    xRoot.size += yRoot.size
  elif xRoot.rank < yRoot.rank:
    xRoot.parent = yRoot
    yRoot.size += xRoot.size
  elif xRoot != yRoot:  # Unless x and y are already in same set, merge them
    yRoot.parent = xRoot
    xRoot.rank = xRoot.rank + 1
    xRoot.size += yRoot.size

似乎修复。