题意:有n个点,m条边,边的权值代表某种颜色,有q个询问,每次询问两点之间是否有相同的颜色将其连通起来,若有,输出有多少种相同的颜色。
分析:Floyd的变形,相当于给出点与点之间的多种关系,然后询问你任意两点之间的关系数目。
代码如下:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
const int M=1010;
const int N=101;
const int INF=1e9;
const double PI=acos(-1.0);
const double Exp=exp(1.0);
int n,m;
int g[N][N][N];
void floyd()
{
int i,j,k,e;
for (k=1;k<=m;k++) // 在每一种颜色下,建立两点之间的关系
for (i=1;i<=n;i++)
for (j=1;j<=n;j++)
for (e=1;e<=n;e++) if (!g[k][j][e])
{
g[k][j][e]=g[k][j][i] & g[k][i][e];
}
}
int main()
{
int i,j;
while(~scanf("%d%d",&n,&m))
{
memset(g,0,sizeof(g));
for (i=0;i<m;i++) {
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
g[c][a][b]=g[c][b][a]= 1;
}
floyd();
int q,a,b;
scanf("%d",&q);
for (i=0;i<q;i++){
int ans=0;
scanf("%d%d",&a,&b);
for (j=1;j<=m;j++) {
if (g[j][a][b] || g[j][b][a]) ans++; // 在某个颜色下,若两点之间有关系(即是连通的),ans++
}
printf("%d\n",ans);
}
}
}