题意:有n个点,m条边,边的权值代表某种颜色,有q个询问,每次询问两点之间是否有相同的颜色将其连通起来,若有,输出有多少种相同的颜色。
分析:Floyd的变形,相当于给出点与点之间的多种关系,然后询问你任意两点之间的关系数目。
#include <bits/stdc++.h> using namespace std; const int maxn = 105; int g[maxn][maxn][maxn]; int n, m; void floyd() { for (int c = 1; c <= m; c++) { for (int k = 1; k <= n; k++) { for (int i = 1; i <= n; i++) { for (int j = 1; j <= n; j++) { if (!g[i][j][c]) { g[i][j][c] = g[i][k][c] & g[k][j][c]; } } } } } } int main() { int a, b, c, q; cin >> n >> m; for (int i = 1; i <= m; i++) { cin >> a >> b >> c; g[a][b][c] = g[b][a][c] = 1; } floyd(); cin >> q; while (q--) { cin >> a >> b; int ans = 0; for (int i = 1; i <= m; i++) { ans += g[a][b][i]; } cout << ans <<endl; } return 0; }