HDOJ 1863 畅通工程

MST，kruskal，并查集，路径压缩；

参考白书的做法：另定义一个边序号数组，排序时按照边权的大小排；

1A，表示已经对WA，TLE，RE等过敏，一提交心里就发毛啊……

代码有点长，重新回到 CB 不太适应。

# include <stdio.h>
# include <stdlib.h>

# define MAXN 105

int n, m;
int w[MAXN], u[MAXN], v[MAXN], r[MAXN], p[MAXN];

int kruskal(void);
int cmp(const void *x, const void *y){return (w[*(int*)x]<w[*(int*)y] ? -1:1);}
int find(int x){return p[x]==x ? x:(p[x] = find(p[x]));}

int main()
{
    int i, ans;

    while (~scanf("%d%d", &n, &m))
    {
        if (!n) break;
        for (i = 1; i <= n; ++i)
        {
            r[i] = i;
            scanf("%d%d%d", &u[i], &v[i], &w[i]);
        }
        if (n < m-1) {puts("?"); continue;}
        for (i = 1; i <= m; ++i) p[i] = i;
        ans = kruskal();
        if (ans > 0) printf("%d\n", ans);
        else puts("?");
    }

    return 0;
}

int kruskal(void)
{
    int i, cost, cnt, x, y, t;

    cnt = 0;
    cost = 0;
    qsort(r+1, n, sizeof(int), cmp);
    for (i = 1; i <= n; ++i)
    {
        t = r[i];
        x = find(u[t]);
        y = find(v[t]);
        if (x != y)
        {
            ++cnt;
            cost += w[t];
            p[x] = y;
        }
    }

    return cnt==m-1 ? cost:-1;
}