最优灌溉

题目链接：http://118.190.20.162/view.page?gpid=T18
试题名称： 最优灌溉
时间限制： 1.0s
内存限制： 256.0MB
问题描述：
问题描述
　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。
输入格式
　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数ai, bi, ci，表示第ai片麦田与第bi片麦田之间可以建立一条水渠，所需要的费用为ci。
输出格式
　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。
样例输入
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≤ci≤10,000。

分析: 最小生成树，取极限数据条件下，图满足 |E|>|V|*log|V| ，即为稠密图，适用prime算法

#include <cstdio>
#include <iostream>
#include <cstring>

using namespace std;
const int VMAX = 1005;
const int MCOST = 10005;

int v, e;
int dis[VMAX];//i到被选中结点集的最小费用
int c[VMAX][VMAX];
bool vis[VMAX];

void Init() {
    for (int i = 1; i <= v; i++) {
        for (int j = 1; j <= v; j++)
            c[i][j] = MCOST;
    }
}
void Read() {
    scanf("%d%d", &v, &e);
    Init(); //对费用矩阵初始化
    int v1,v2,cost;
    for (int i = 1; i <= e; i++) {
        scanf("%d%d%d", &v1, &v2, &cost);
        c[v1][v2] = c[v2][v1] = cost;
    }
}

void Prime(int s) {
    int res = 0;
    //init
    memset(vis, false, sizeof(vis));
    vis[s] = true;
    for (int i = 1; i <= v; i++)
        if(i != s)
            dis[i] = c[s][i];

    int new_v;
    for (int i = 1; i < v; i++) { //选(v-1)条边
        int minDist = MCOST;
        for (int j = 1; j <= v; j++) { //从当前未选定的结点中找到距已选定结点集合中最小费用的点
            if (!vis[j] && dis[j] < minDist) {
                new_v = j;
                minDist = dis[j];
            }
        }
        vis[new_v] = true;
        res += dis[new_v];
        for (int j = 1; j <= v; j++) { //尝试更新dis(区别于dijkstra:c[new_v][j]+dis[new_v]<dis[j])
            if (!vis[j] && c[new_v][j] < dis[j]) {
                dis[j] = c[new_v][j];
            }
        }
    }
    printf("%d\n", res);
}

int main() {
    Read();
    Prime(1); //水井在第一个麦田处
    return 0;
}