WHU 1542 Countries (floyd)

题意：

　　在小明出生的星球X上有n国家。

　　一些国家通过结盟而拥有良好的双边关系，因此他们的公民得益于这些政策，使得所有这些国家之间的旅行变得免费。

　　但是，不同联盟之间的旅行就不是这么容易了。如果可能，它必须花一些钱。

　　请帮小明计算出国家之间的最小花费。

　　输入：

　　　　输入包含多组样例。

　　　　每组样例的第一行有两个整数 n 和 m 。(1<=n<=10^5, 1<=m<=10^5)

　　　　接下来的m行，每行有x, y, c。 其中 c 表示 x, y之间的花费。如果c为0，则表示x和y属于同一个联盟。(1<=x, y<=n, 0<=c<=10^9)

　　　　你可以假设两个国家之间没有多于一条路。

　　　　然后下一行是一个整数q，代表有多少个询问。(1<=q<=200)

　　　　接下来有q行，每行包括x, y。(1<=x, y<=n)

　　　　保证不会超过两百个联盟。

　　　　当输入n=0时，结束。

　　输出：

　　　　对于每组输入，每个询问输出一行。如果不可能，输出"-1"。

思路：

　　对于同一个联盟的国家，我们可以把他们看成一个国家，因为他们一定是连起来的。 然后跑floyd。

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

using namespace std;

typedef long long ll;

const int MAXN = (int) 1e5+7;
const int MAXC = 205;

struct Edge{
    int x, y;
    ll c;
}a[MAXN];

int p[MAXN];
int f[MAXN];
ll d[MAXC][MAXC];
int n, m, q;
int cnt;

int Find(int x) {
    int res = x;
    int t;

    while (f[res]!=res) res = f[res];
    while (x!=res) {
        t = x;
        x = f[x];
        f[t] = res;
    }
    return res;
}

void Union(int x, int y) {
    f[Find(x)] = Find(y);
}

void floyd() {
    for (int k = 0; k < cnt; k++)
        for (int i = 0; i < cnt; i++)
            for (int j = 0; j < cnt; j++)
                if (-1!=d[i][k] && -1!=d[k][j]) {
                    if (-1==d[i][j]) {
                        d[i][j] = d[i][k]+d[k][j];
                    } else
                        d[i][j] = min(d[i][j], d[i][k]+d[k][j]);
                }
}

int main() {
    #ifdef Phantom01
        freopen("WHU1542.txt", "r", stdin);
    #endif // Phantom01

    while (scanf("%d", &n)!=EOF) {
        if (0==n) break;
        scanf("%d", &m);
        cnt = 0;
        for (int i = 1; i <= n; i++)
            f[i] = i;
        for (int i = 0; i < m; i++) {
            scanf("%d%d%lld", &a[i].x, &a[i].y, &a[i].c);
            if (0 == a[i].c) Union(a[i].x, a[i].y);
        }
        for (int i = 1; i <= n; i++) {
            if (i==f[i]) //如果是根，
                p[i] = cnt++;
        }
        for (int i = 0; i < cnt; i++) {
            for (int j = 0; j < cnt; j++)
                d[i][j] = -1;
            d[i][i] = 0;
        }
        int x, y;
        for (int i = 0; i < m; i++) {
            x = p[Find(a[i].x)];
            y = p[Find(a[i].y)];
            d[x][y] = -1==d[x][y] ? a[i].c : min(d[x][y], a[i].c);
            d[y][x] = -1==d[y][x] ? a[i].c : min(d[y][x], a[i].c);
        }
        floyd();
        scanf("%d", &q);
        for (int i = 0; i < q; i++) {
            scanf("%d%d", &x, &y);
            printf("%lld
", d[p[Find(x)]][p[Find(y)]]);
        }
    }

    return 0;
}