Codeforces400D Dima and Bacteria

题意：给你一个无向有权的图，图上的点被分成了几类，对于同类的点你需要判断它们之间相互的最短距离是不是0.满足这个条件之后要输出的是类与类之间的最短距离的矩阵。点给到10^5这么多，判断同类的点显然不能跑最短路，所以直接的方法必然是并查集，对边为0的点做一次并查集，对同类的点判一下find(x)==find(y)就可以了。 然后就是将同类的点抽象出一个新的点，这个时候只有500个点，然后就可以跑一下floyd了。 题意有坑的地方，所以没有AC比赛的时候。 这题倒是很好的练了一下基础的内容并查集和floyd。

#pragma warning(disable:4996)

#include<iostream>

#include<cstring>

#include<cstdio>

#include<algorithm>

#include<vector>

#include<cmath>

#include<queue>

#define ll long long

#define maxn 100500

#define inf 0x3f3f3f3f

using namespace std;



vector<int> ZG[maxn];

int type[550];

int sum[550];

int n, m, k;





int fa[maxn];

int find(int x)

{

	return fa[x] == x ? fa[x] : fa[x] = find(fa[x]);

}



bool deterMine(int index)

{

	int sid = sum[index - 1];

	for (int i = sid + 1; i <= sid + type[index]; i++){

		if (find(sid + 1) != find(i)) return false;

	}

	return true;

}



int dist[550][550];



int main()

{

	while (cin >> n >> m >> k)

	{

		sum[0] = 0;

		for (int i = 1; i <= k; i++){

			scanf("%d", &type[i]);

			sum[i] = sum[i - 1] + type[i];

		}

		int ui, vi, wi;

		memset(dist, 0x3f, sizeof(dist));

		for (int i = 0; i <= n; i++) fa[i] = i;

		for (int i = 0; i <= k; i++) { dist[i][i] = 0; }

		for (int i = 0; i < m; i++){

			scanf("%d%d%d", &ui, &vi, &wi);

			int ut = lower_bound(sum, sum + 1 + k, ui) - sum;

			int	vt = lower_bound(sum, sum + 1 + k, vi) - sum;

			dist[ut][vt] = min(dist[ut][vt], wi);

			dist[vt][ut] = min(dist[vt][ut], wi);

			if (wi == 0) {

				if (find(ui) != find(vi)){

					fa[find(ui)] = fa[find(vi)];

				}

			}

		}

		for (int i = 1; i <= n; i++) find(i);

		bool flag = true;

		for (int i = 1; i <= k; i++){

			if (deterMine(i) == false){

				flag = false;

			}

		}

		if (!flag) { puts("No"); continue; }

		puts("Yes");

		for (int kk = 1; kk <= k; kk++){

			for (int i = 1; i <= k; i++){

				for (int j = 1; j <= k; j++){

					if (dist[i][kk] + dist[kk][j] < dist[i][j]){

						dist[i][j] = dist[i][kk] + dist[kk][j];

					}

				}

			}

		}

		for (int i = 1; i <= k; i++){

			for (int j = 1; j <= k; j++){

				if (j != 1) printf(" ");

				if (dist[i][j] < inf) printf("%d", dist[i][j]);

				else printf("%d", -1);

			}

			puts("");

		}

	}

	return 0;

}