Question

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

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

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

Answer 1

您可以使用名为disjoint-set的数据结构（here是一个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

我认为计数将在provided answer中关闭。例如。如果行更改为rows = [[1, 0, 0], [1, 1, 1], [1, 0, 0]]仍然获得4，但它应该是5.将Union更改为

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

似乎已经解决了。

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

2 个答案: