技术图文:如何利用C# 实现 Kruskal 最小生成树算法?

背景

以前我写过一些图文来介绍有关数据结构与算法的知识:

本次,向大家介绍图论中构造最小生成树的 Kruskal 算法。


技术分析

Kruskal 算法:

Krusal算法

例子:

在这里插入图片描述

该例子演示了一个含有6个结点,10条边的连通网,通过 Kruskal 算法逐步演化为含有6个结点,5条边的连通子网的过程,即构造最小生成树的过程。


代码实现

Step1 构造边表结点的结构 EdgeNode

public class EdgeNode
{
    /// <summary>
    /// 获取边终点在顶点数组中的位置
    /// </summary>
    public int Index { get; }

    /// <summary>
    /// 获取边上的权值
    /// </summary>
    public double Weight { get; }

    /// <summary>
    /// 获取或设置下一个邻接点
    /// </summary>
    public EdgeNode Next { get; set; }

    /// <summary>
    /// 初始化EdgeNode类的新实例
    /// </summary>
    /// <param name="index">边终点在顶点数组中的位置</param>
    /// <param name="weight">边上的权值</param>
    /// <param name="next">下一个邻接点</param>
    public EdgeNode(int index, double weight = 0.0, EdgeNode next = null)
    {
        if (index < 0)
            throw new ArgumentOutOfRangeException();

        Index = index;
        Weight = weight;
        Next = next;
    }
}

Step2 构造顶点表结点的结构 VertexNode

public class VertexNode
{
    /// <summary>
    /// 获取或设置顶点的名字
    /// </summary>
    public string VertexName { get; set; }

    /// <summary>
    /// 获取或设置顶点是否被访问
    /// </summary>
    public bool Visited { get; set; }

    /// <summary>
    /// 获取或设置顶点的第一个邻接点
    /// </summary>
    public EdgeNode FirstNode { get; set; }

    /// <summary>
    /// 初始化VertexNode类的新实例
    /// </summary>
    /// <param name="vName">顶点的名字</param>
    /// <param name="firstNode">顶点的第一个邻接点</param>
    public VertexNode(string vName, EdgeNode firstNode = null)
    {
        VertexName = vName;
        Visited = false;
        FirstNode = firstNode;
    }
}

Step3 构造利用邻接表存储图的结构AdGraph

通过 AdGraph 的索引器可以为顶点表赋值,通过 AddEdge 方法可以为边表赋值。

public class AdGraph
{
    private readonly VertexNode[] _vertexList; //结点表

    /// <summary>
    /// 获取图的结点数
    /// </summary>
    public int VertexCount { get; }

    /// <summary>
    /// 初始化AdGraph类的新实例
    /// </summary>
    /// <param name="vCount">图中结点的个数</param>
    public AdGraph(int vCount)
    {
        if (vCount <= 0)
            throw new ArgumentOutOfRangeException();

        VertexCount = vCount;
        _vertexList = new VertexNode[vCount];
    }

    /// <summary>
    /// 获取或设置图中各结点的名称
    /// </summary>
    /// <param name="index">结点名称从零开始的索引</param>
    /// <returns>指定索引处结点的名称</returns>
    public string this[int index]
    {
        get
        {
            if (index < 0 || index > VertexCount - 1)
                throw new ArgumentOutOfRangeException();

            return _vertexList[index] == null
                ? "NULL"
                : _vertexList[index].VertexName;
        }
        set
        {
            if (index < 0 || index > VertexCount - 1)
                throw new ArgumentOutOfRangeException();

            if (_vertexList[index] == null)
                _vertexList[index] = new VertexNode(value);
            else
                _vertexList[index].VertexName = value;
        }
    }

    /// <summary>
    /// 得到结点在结点表中的位置
    /// </summary>
    /// <param name="vertexName">结点的名称</param>
    /// <returns>结点的位置</returns>
    private int GetIndex(string vertexName)
    {
        int i;
        for (i = 0; i < VertexCount; i++)
        {
            if (_vertexList[i] != null && _vertexList[i].VertexName == vertexName)
                break;
        }
        return i == VertexCount ? -1 : i;
    }

    /// <summary>
    /// 给图加边
    /// </summary>
    /// <param name="startVertexName">起始结点的名字</param>
    /// <param name="endVertexName">终止结点的名字</param>
    /// <param name="weight">边上的权值</param>
    public void AddEdge(string startVertexName, string endVertexName
        , double weight = 0.0)
    {
        int i = GetIndex(startVertexName);
        int j = GetIndex(endVertexName);

        if (i == -1 || j == -1)
            throw new Exception("图中不存在该边.");

        EdgeNode temp = _vertexList[i].FirstNode;
        if (temp == null)
        {
            _vertexList[i].FirstNode = new EdgeNode(j, weight);
        }
        else
        {
            while (temp.Next != null)
                temp = temp.Next;
            temp.Next = new EdgeNode(j, weight);
        }
    }
}    

上面例子对应的邻接表如下所示:

邻接表

Step4 构造最小生成树结点的结构 SpanTreeNode

public class SpanTreeNode
{
    /// <summary>
    /// 获取或设置结点本身的名称
    /// </summary>
    public string SelfName { get; }

    /// <summary>
    /// 获取或设置结点双亲的名称
    /// </summary>
    public string ParentName { get; }

    /// <summary>
    /// 获取或设置边的权值
    /// </summary>
    public double Weight { get; set; }

    /// <summary>
    /// 构造SpanTreeNode实例
    /// </summary>
    /// <param name="selfName">结点本身的名称</param>
    /// <param name="parentName">结点双亲的名称</param>
    /// <param name="weight">边的权值</param>
    public SpanTreeNode(string selfName, string parentName, double weight)
    {
        if (string.IsNullOrEmpty(selfName) || string.IsNullOrEmpty(parentName))
            throw new ArgumentNullException();

        SelfName = selfName;
        ParentName = parentName;
        Weight = weight;
    }
}

Step5 构造边的结构 Edge

internal class Edge
{
    /// <summary>
    /// 起点编号
    /// </summary>
    public int Begin { get;}

    /// <summary>
    /// 终点编号
    /// </summary>
    public int End { get; }

    /// <summary>
    /// 权值
    /// </summary>
    public double Weight { get; }

    /// <summary>
    /// 创建一个 Edge 类的新实例
    /// </summary>
    /// <param name="begin">起点编号</param>
    /// <param name="end">终点编号</param>
    /// <param name="weight">权值</param>

    public Edge(int begin, int end, double weight = 0.0)
    {
        Begin = begin;
        End = end;
        Weight = weight;
    }
}

Step6 获取边集合的方法 GetEdges

private Edge[] GetEdges()
{
    for (int i = 0; i < VertexCount; i++)
        _vertexList[i].Visited = false;

    List<Edge> result = new List<Edge>();

    for (int i = 0; i < VertexCount; i++)
    {
        _vertexList[i].Visited = true;
        EdgeNode p = _vertexList[i].FirstNode;
        while (p != null)
        {
            if (_vertexList[p.Index].Visited == false)
            {
                Edge edge = new Edge(i, p.Index, p.Weight);
                result.Add(edge);
            }
            p = p.Next;
        }
    }
    return result.OrderBy(a => a.Weight).ToArray();
}

上面例子对应的边的集合如下所示:

Step7 获取最小生成树的 Kruskal 算法。

private int Find(int[] parent, int f)
{
    while (parent[f] > 0)
        f = parent[f];
    return f;
}

/// <summary>
/// 克鲁斯卡尔算法 最小生成树
/// </summary>
/// <returns></returns>
public SpanTreeNode[] MiniSpanTree()
{
    int[] parent = new int[VertexCount];
    for (int i = 0; i < VertexCount; i++)
    {
        parent[i] = 0;
    }
    SpanTreeNode[] tree = new SpanTreeNode[VertexCount];
    int count = 0;
    Edge[] edges = GetEdges();

    for (int i = 0; i < edges.Length; i++)
    {
        int begin = edges[i].Begin;
        int end = edges[i].End;
        int n = Find(parent, begin);
        int m = Find(parent, end);
        if (n != m)
        {
            if (i == 0)
            {
                tree[count] = new SpanTreeNode(_vertexList[begin].VertexName, "NULL", 0.0);
                count++;
            }
            parent[n] = m;
            tree[count] = new SpanTreeNode(_vertexList[end].VertexName,
                _vertexList[begin].VertexName, edges[i].Weight);
            count++;
        }
    }
    return tree;
}

总结

到此为止代码部分就全部介绍完了,我们来看一下上面例子的应用。

static void Main(string[] args)
{
    AdGraph alg = new AdGraph(6);
    alg[0] = "V0";
    alg[1] = "V1";
    alg[2] = "V2";
    alg[3] = "V3";
    alg[4] = "V4";
    alg[5] = "V5";
    alg.AddEdge("V0", "V1", 6);
    alg.AddEdge("V0", "V2", 1);
    alg.AddEdge("V0", "V3", 5);
    alg.AddEdge("V1", "V0", 6);
    alg.AddEdge("V1", "V2", 5);
    alg.AddEdge("V1", "V4", 3);
    alg.AddEdge("V2", "V0", 1);
    alg.AddEdge("V2", "V1", 5);
    alg.AddEdge("V2", "V3", 7);
    alg.AddEdge("V2", "V4", 5);
    alg.AddEdge("V2", "V5", 4);
    alg.AddEdge("V3", "V0", 5);
    alg.AddEdge("V3", "V2", 7);
    alg.AddEdge("V3", "V5", 2);
    alg.AddEdge("V4", "V1", 3);
    alg.AddEdge("V4", "V2", 5);
    alg.AddEdge("V4", "V5", 6);
    alg.AddEdge("V5", "V2", 4);
    alg.AddEdge("V5", "V3", 2);
    alg.AddEdge("V5", "V4", 6);
    SpanTreeNode[] tree = alg.MiniSpanTree();
    double sum = 0;
    for (int i = 0; i < tree.Length; i++)
    {
        string str = "(" + tree[i].ParentName + ","
                + tree[i].SelfName + ") Weight:"
                + tree[i].Weight;
        Console.WriteLine(str);
        sum += tree[i].Weight;
    }
    Console.WriteLine(sum);
}
结果

我们再通过一个例子来演示如何应用:

地图

上面是一幅纽约市附近的地图,对应的数据存储在 graph.txt 文件中。

数据

读入该文件,构造好 AdGraph 结构后,调用我们写好的 Kruskal 算法,得到的结果如下:

结果

是不是很有趣,今天就到这里吧!马上要放假了,我们的招新活动也即将开启,希望大家关注呦!

See You!


相关图文

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,558评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,002评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,036评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,024评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,144评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,255评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,295评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,068评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,478评论 1 305
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,789评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,965评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,649评论 4 336
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,267评论 3 318
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,982评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,223评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,800评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,847评论 2 351

推荐阅读更多精彩内容