图结构练习——最小生成树

图结构练习——最小生成树

Time Limit: 1000MS Memory limit: 65536K

题目描述

 有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。

 

输入

 输入包含多组数据，格式如下。

第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)

剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。

 

输出

 每组输出占一行，仅输出最小花费。

示例输入

3 2
1 2 1
1 3 1
1 0

示例输出

2
0

提示

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
using namespace std;
struct node
{
    int u;
    int v;
    int w;
}b[15000];
int a[150];

bool cmp(struct node a, struct node b)
{
    return a.w < b.w;
}
int f(int x)
{
    while(x != a[x]){
        x = a[x];
    }
    return x;
}
int mer(int x, int y)
{
    int f1, f2;
    f1 = f(x);
    f2 = f(y);
    if(f1 != f2){
        a[f1] = f2;
        return 1;
    }else {
        return 0;
    }
}
int main()
{
        int n, m;
        int cnt;
        int sum;
        while(~scanf("%d %d", &n, &m)){//读入n和m，n表示定点个数，m表示边的条数
            cnt = 0;
            sum = 0;
            int i;
            for(i = 0;i < m;i++){
                scanf("%d %d %d", &b[i].u, &b[i].v, &b[i].w);
            }
            sort(b,b+m,cmp);// 按权值从小到大对边进行排序
            for(i  =1;i <= n;i++){//并查集初始化
                a[i] = i;
            }
            for(i = 0;i < m;i++){ // Kruskal算法的核心，开始从小到大枚举每条边
                if(mer(b[i].u,b[i].v)){//判断一条边的两个顶点是否已经连通，即判断是否在同一个集合中，如果连通，选用这条边
                    sum += b[i].w;
                    cnt++;
                }
                if(cnt == n-1){//直到选用了n-1条边之后退出
                    break;
                }
            }
            printf("%d\n", sum);
        }

        return 0;
}

prim的最小生成树

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define inf 0x3f3f3f3f

int dis[150];
int book[150];
int a[150][150];

int main()
{
    int n, m;
    while(~scanf("%d %d", &n, &m)){
        memset(book,0,sizeof(book));
        memset(dis,0,sizeof(dis));
        memset(a,0,sizeof(a));
        int i, j, k;
        for(i = 1;i <= n;i++){
            for(j = 1;j <= n;j++){
                if(i == j){
                    a[i][j] = 0;
                }else {
                    a[i][j] = inf;
                }
            }
        }
        int u, v, w;
        for(i = 0;i < m;i++){
            scanf("%d %d %d", &u, &v, &w);
            if(a[u][v] > w){
                a[u][v] = w;
                a[v][u] = w;
            }
        }
        for(i = 1;i <= n;i++){//初始化dis数组，这里是1号顶点到各个顶点的初始距离，因为当前生成树中只有1号定点
            dis[i] = a[1][i];
        }
        book[1] = 1;//将1号定点加入到最小生成树中去，这里用book来标记是否加入到生成树
        int sum = 0;
        int count = 1;
        int min;
        while(count < n){
            min = inf;
            for(i = 1;i <= n;i++){
                if(book[i] == 0 && dis[i] < min){
                    min = dis[i];
                    j = i;
                }
            }
            book[j] = 1;
            sum += min;
            count++;
            for(k = 1;k <= n;k++){ //扫描当前j点所有的边，再以j为中间点，更新生成树到每一个非树顶点的距离
                if(book[k] == 0 && dis[k] > a[j][k]){
                    dis[k] = a[j][k];
                }
            }
        }
        printf("%d\n", sum);
    }

    return 0;
}