最优灌溉 (201412-4)

问题描述

　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。

输入格式

　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数a _i, b _i, c _i，表示第a _i片麦田与第b _i片麦田之间可以建立一条水渠，所需要的费用为c _i。

输出格式

　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。

样例输入

4 4
1 2 1
2 3 4
2 4 2
3 4 3

样例输出

6

样例说明

　　建立以下三条水渠：麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。

评测用例规模与约定

　　前20%的评测用例满足：n≤5。
　　前40%的评测用例满足：n≤20。
　　前60%的评测用例满足：n≤100。
　　所有评测用例都满足：1≤n≤1000，1≤m≤100,000，1≤c _i≤10,000。

思路：读一遍题就发现，这是一道基本的最小生成树的例子。用Prim算法 或者 Kruskal算法。 注意map数组初始化时将 自己到自己的距离设为0，没有路径达到的点 距离设为MAXX。 本题不难，用基本算法做也没超时。

#include <iostream>
#include <cstring>
using namespace std;

#define MAX 1010
#define MAXX 10010

bool vis[MAX];
int dis[MAX];
int map[MAX][MAX];
int n,m;

void Prim(){
    memset(vis,0,sizeof(vis));
    int i,j,k = 0;
    for(i=1;i<=n;i++)
        dis[i] = map[1][i];
    for(i=1;i<=n;i++){
        int temp = MAXX;
        for(j=1;j<=n;j++){
            if(!vis[j] && dis[j]<temp){
                temp = 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];
        }
    }
    for(i=2;i<=n;i++)
        dis[1] += dis[i];
    cout<<dis[1]<<endl;
}

int main(){
    cin>>n>>m;
    int i,j;
    for(i=0;i<=n;i++){
        for(j=0;j<=n;j++){
            if(i == j)
                map[i][j] = 0;
            else
                map[i][j] = MAXX;
        }
    }
    for(i=0;i<m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        map[a][b] = map[b][a] = c;
    }
    Prim();
    return 0;
}