最短网络(agrinet)

线上OJ：

【例4-11】最短网络(agrinet)

核心思想：（使用 Kruskal 方法求最小生成树）

算法描述:

1、初始化并查集。p[x]=x。
2、初始化输出结果ans=0
3、将所有边从小到大排序（sort）。
4、计数器 k=0，记载当前找到几条边;
5、核心代码如下

for (i=1; i<=cnt; i++)      // cnt为边的数量。循环所有已从小到大排序的边
if  这是一条u,v不属于同一集合的边(u,v)(因为已经排序，所以必为最小)，
begin
　 ①合并u,v所在的集合，相当于把边(u,v)加入最小生成树。
　    ②ans += W(u,v)
③k++
④如果k=n-1,说明最小生成树已经生成，则break;
end;

6、结束，ans即为最小生成树的总权值之和。

#include <bits/stdc++.h>
using namespace std;

const int N = 10010;

struct Node
{
    int x, y, v;    // 结构体存储边的两端坐标x和y，以及边权v
};
Node e[N];
int n, ans, cnt = 0;  // cnt 记载边的数量
int p[N]; // p[i]表示 i 的根节点ok Ok.bool cmp(Node a, Node b)

bool cmp(Node a, Node b)
{
    return a.v < b.v;
}

int find(int x)  // 找根节点
{
    if(p[x] != x)  p[x] = find(p[x]);
    return p[x];
}

void unionn(int x, int y)  // 合并x和y
{
    int a = find(x), b = find(y);
    if(a != b)  p[a] = b;
}

int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++)
        {
            int a;
            cin >> a;
            if(a != 0)  // 存储权不为0 的边
            {
                e[++cnt].v = a;  // 存储边的权值，cnt 为边的数量
                e[cnt].x = i;  // x为边的起始端点
                e[cnt].y = j;  // y为边的结束端点
            }
        }

    for(int i = 1; i <= n; i++) p[i] = i;  // 初始化每个元素的根节点为自己（即每个元素都是独立的）

    sort(e + 1, e + 1 + cnt, cmp);  // e为结构体，根据 e[i].v 的值进行升序排序

    int k = 0;
    for(int i = 1; i <= cnt; i++)  // 总共有cnt条边，按边权从小到大的顺序，寻找第一个两边端点分属不同集合的边，将他们合并
    {
        if( find(e[i].x) != find(e[i].y) )  // 如果边的两个端点在不同的阵营
        {
            unionn(e[i].x, e[i].y);   // 则合并
            ans += e[i].v;  // 累加输出结果
            k++;  // 当前找到了几条边
            if( k == n-1 )  break;  // 如果边的数量达到n-1，则跳出循环
        }
    }

    cout << ans << endl;
    return 0;
}