从对列表中创建邻接列表类型结构

时间:2013-06-28 00:32:18

标签: c# algorithm dictionary adjacency-list connected-components

在C#中,我有

class Pair{

  int val1;
  int val2;
}

我有一个来自源的对列表: -

List<Pair> sList = new List<Pair>();

   1 | 2
   2 | 3
   1 | 4
   4 | 6

我需要将其转换为以下类型的结构: -

 [1, [2, 3, 4, 6]]  
 [2, [3]]
 [3, [2]]
 [4, [1,6]]
 [6, [4]]

最好的方法是什么(没有LINQ)?

2 个答案:

答案 0 :(得分:5)

我会选择ILookup<int, int>,但您还需要包含反向关联:

var result = sList.Union(sList.Select(p => new Pair { val1 = p.val2, val2 = p.val1 }))
                  .ToLookup(p => p.val1, p => p.val2);

如果没有Linq,你可以得到类似的结果:

var dict = new Dictionary<int, List<int>>();
foreach(var pair in sList)
{
    if (!dict.ContainsKey(pair.val1))
    {
        dict[pair.val1] = new List<int>();
    }
    if (!dict.ContainsKey(pair.val2))
    {
        dict[pair.val2] = new List<int>();
    }

    dict[pair.val1].Add(pair.val2);
    dict[pair.val2].Add(pair.val1);
}

上述两种方法都会生成Adjacency List,但是根据您的评论,您会更喜欢 Connected Component Labeling

var groups = new List<HashSet<int>>();
foreach (var p in sList)
{
    var merge = new List<HashSet<int>>();
    foreach(var g in groups)
    {
        if (g.Contains(p.val1) || g.Contains(p.val2))
        {
            merge.Add(g);
        }
    }

    if (merge.Count == 0)
    {
        var h = new HashSet<int>();
        groups.Add(h);
        merge.Add(h);
    }

    merge[0].Add(p.val1);
    merge[0].Add(p.val2);
    for(int i = 1; i < merge.Count; i ++)
    {
        foreach(int v in merge[i])
        {
            merge[0].Add(v);
        }

        groups.Remove(merge[i]);
    }
}

当输入

sList = 
    1 | 2
    4 | 6
    2 | 3
    1 | 4
    9 | 10

这将产生输出:

groups = 
    [ 1, 2, 3, 4, 6 ]
    [ 9, 10 ]

将其转换为您想要的格式并不太难:

var dict = new Dictionary<int, List<int>>();
foreach(var g in groups)
{
    foreach(var v in g)
    {
        var list = new List<int>(g);
        list.Remove(g);
        dict.Add(v, list)
    }
}

使用前面的例子:

dict =
    1 | [ 2, 3, 4, 6 ]
    2 | [ 1, 3, 4, 6 ]
    3 | [ 1, 2, 4, 6 ]
    4 | [ 1, 2, 3, 6 ]
    6 | [ 1, 2, 3, 4 ]
    9 | [ 9 ]
    10 | [ 10 ]

答案 1 :(得分:1)

您可以使用LINQ的GroupBy方法执行此操作,如下所示:

var adj = sList
    .GroupBy(p => p.val1)
    .ToDictionary(g => g.Key, g => g.Select(p => p.val2).ToList());

请注意,这不会计算图表的传递闭包,即只会显示直接链接。

在.NET 4及更高版本中,您也可以使用Tuple<int,int>而不是自己的Pair课程。