[HDOJ1863]畅通工程

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1863

新学习了prim算法，把之前的dijkstra改了改，d数组用来存上一个距离当前节点最近的长度，而不是源点到当前最近的长度就可以了。初始化d为inf，如果条件不够那和算出来必定>inf(没考虑溢出，因为我定义的inf并不是很大)（题目小坑，把n和m输入顺序反了一下。）

#pragma warning(disable:4996)


#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <fstream>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>

using namespace std;

const int maxn = 105;
const int inf = 0xffffff;
int d[maxn];
int G[maxn][maxn];
int vis[maxn];
int n, m;

void init() {
    memset(vis, 0, sizeof(vis));
    for(int i = 0; i <= n; i++) {
        d[i] = inf;
        for(int j = 0; j <= n; j++) {
            G[i][j] = G[j][i] = inf;
        }
        G[i][i] = 0;
    }
}

int prim(int start) {
    d[start] = 0;
    for(int i = 1; i <= n; i++) {
        int u = -1;
        for(int j = 1; j <= n; j++) {
            if(!vis[j]) {
                if(u == -1 || d[j] < d[u]) {
                    u = j;
                }
            }
        }
        vis[u] = 1;
        for(int j = 1; j <= n; j++) {
            if(!vis[j]) {
                d[j] = min(G[u][j], d[j]);
            }
        }
    }
    int sp = 0;
    for(int i = 1; i <= n; i++) {
        sp += d[i];
    }
    return sp;
}

int main() {
    // freopen("in", "r", stdin);
    int u, v, w;
    while(~scanf("%d %d", &m, &n) && m) {
        init();
        while(m--) {
            scanf("%d %d %d", &u, &v, &w);
            if(w < G[u][v]) {
                G[u][v] = G[v][u] = w;
            }
        }
        int len = prim(1);
        if(len > inf) puts("?");
        else printf("%d
", len);
    }
}