HDOJ 1875 畅通工程再续

示例的结果为1414.0,kruskal的返回类型写成了int,查了好几遍都没发现问题;

还是MST,kruskal,并查集,路径压缩;

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

# define MAXN 105

typedef struct
{
    int u, v;
    double d;
} Bridge;

int m;
int p[MAXN], a[MAXN][2];
Bridge b[MAXN*50];

double kruskal(int n);
int find(int x)
{
    return x==p[x] ? x:(p[x] = find(p[x]));
}
int cmp(const void*x, const void*y)
{
    return (*(Bridge*)x).d>(*(Bridge*)y).d ? 1:-1;
}

int main()
{
    double t, ans;
    int T, i, j, k, dx, dy;

    scanf("%d", &T);
    while (T--)
    {
        scanf("%d", &m);
        for (i = 1; i <= m; ++i)
        {
            p[i] = i;
            scanf("%d%d", &a[i][0], &a[i][1]);
        }

        k = 0;
        for (i = 1; i < m; ++i)
            for (j = i+1; j <= m; ++j)
            {
                dx = a[i][0] - a[j][0];
                dy = a[i][1] - a[j][1];
                t = sqrt(dx*dx + dy*dy);
                if (t>=10 && t<=1000)
                {
                    ++k;
                    b[k].u = i;
                    b[k].v = j;
                    b[k].d = 100*t;
                }
            }

        ans = kruskal(k);
        if (ans >= 0) printf("%.1lf\n", ans);
        else puts("oh!");
    }

    return 0;
}

double kruskal(int n)
{
    double cost;
    int i, x, y, cnt;

    qsort(b+1, n, sizeof(b[0]), cmp);

    cnt = 0;
    cost = 0;
    for (i = 1; i <= n; ++i)
    {
        x = find(b[i].u);
        y = find(b[i].v);
        if (x != y)
        {
            ++cnt;
            cost += b[i].d;
            p[x] = y;
        }
    }

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

1A,还很荣幸地登上了GCC的第一版,有图为证:

事实上,这道题用GCC提交的总共就这18个……

原文地址:https://www.cnblogs.com/JMDWQ/p/2464917.html