NYOJ 38 布线问题 最小生成树 prim与Kruskal

布线问题
时间限制：1000 ms | 内存限制：65535 KB
难度：4
描述
南阳理工学院要进行用电线路改造，现在校长要求设计师设计出一种布线方式，该布线方式需要满足以下条件：
1、把所有的楼都供上电。
2、所用电线花费最少
输入
第一行是一个整数n表示有n组测试数据。(n<5)
每组测试数据的第一行是两个整数v,e.
v表示学校里楼的总个数(v<=500)
随后的e行里，每行有三个整数a,b,c表示a与b之间如果建铺设线路花费为c(c<=100)。（哪两栋楼间如果没有指明花费，则表示这两栋楼直接连通需要费用太大或者不可能连通）
随后的1行里，有v个整数,其中第i个数表示从第i号楼接线到外界供电设施所需要的费用。( 0 < e < v*(v-1)/2 )
（楼的编号从1开始），由于安全问题，只能选择一个楼连接到外界供电设备。
数据保证至少存在一种方案满足要求。
输出
每组测试数据输出一个正整数,表示铺设满足校长要求的线路的最小花费。
样例输入
1
4 6
1 2 10
2 3 10
3 1 10
1 4 1
2 4 1
3 4 1
1 3 5 6
样例输出
4

[分析]
比较简单的最小生成树问题。
直接用prim或kruskal做，然后取最小 外界供电设备 相加得ans。

[代码]

Kruskal

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

#define MAX 505
struct UnionFind
{
    int fa[MAX];
    int num[MAX];
    UnionFind()
    {
        make_set();
    }
    void make_set()//初始化
    {
        for (int i = 0; i < MAX; i++)
        {
            fa[i] = i;
            num[i] = 0;
        }
    }

    int find_set(int x)//查
    {
        return x == fa[x] ? x : fa[x] = find_set(fa[x]);
    }

    void union_set(int a, int b)//合并
    {
        int x = find_set(a);
        int y = find_set(b);
        if (x != y)
        {
            fa[y] = x;
            num[x] += num[y];
        }
    }
};

struct Node
{
    int a, b, c;
}node[150000];



bool cmp(Node A, Node B)
{
    return A.c < B.c;
}
int main()
{
    int t;
    scanf("%d", &t);
    while (t--)
    {
        UnionFind uf;
        int v, e;
        scanf("%d%d", &v, &e);
        for (int i = 0;i < e;i++)
        {
            scanf("%d%d%d", &node[i].a, &node[i].b, &node[i].c);
        }
        sort(node, node + e, cmp);

        int minn = 0x3f3f3f3f;
        int x;
        for (int i = 0;i < v;i++)
        {
            scanf("%d", &x);
            minn = min(minn, x);
        }

        int ans=0;
        for (int i = 0;i < e;i++)
        {
            if (uf.find_set(node[i].a)!= uf.find_set(node[i].b))
            {
                uf.union_set(node[i].a, node[i].b);
                ans += node[i].c;
                if (uf.num[uf.find_set(node[i].a)]==v)break;
            }
        }
        printf("%d\n", ans + minn);
    }
}

prim

#include<cstdio>
#include<cstring>
#define inf 0x3f3f3f3f
int n;
int dis[505];
int vis[505];
int map[505][505];
void Prim() {
    int i, j, k, tmp;
    for (i = 1;i <= n;i++)//n为楼数
    {
        dis[i] = inf;//初始化
    }
    dis[1] = 0;
    for (i = 1;i <= n;i++) 
    {
        tmp = inf;
        for (j = 1;j <= n;j++) 
        {
            if (!vis[j] && tmp>dis[j]) 
            {
                tmp = dis[j];
                k = j;
            }//找出最小距离的节点   
        }
        vis[k] = 1;//把访问的节点做标记  
        for (j = 1;j <= n;j++) {
            if (!vis[j] && dis[j]>map[k][j])
                dis[j] = map[k][j];//更新最短距离   
        }
    }
}
int main()
{
    int t;
    int a, b, c;
    scanf("%d", &t);
    while (t--)
    {
        int v, e;
        memset(vis, 0, sizeof(vis));
        memset(map, inf, sizeof(map));
        scanf("%d%d", &v, &e);
        n = v;
        for (int i = 0;i < e;i++)
        {
            scanf("%d%d%d", &a, &b, &c);
            map[a][b] = map[b][a] = c;
        }
        int minn = inf;
        int x;
        for (int i = 0;i < v;i++)
        {
            scanf("%d", &x);
            if (x < minn)minn = x;
        }

        Prim();
        int ans = 0;
        for (int i = 1;i <= v;i++)
        {
            ans += dis[i];
        }

        printf("%d\n", ans + minn);
    }
}