Codeforces Round #286 (Div. 2) B. Mr. Kitayuta Colorful Graph（并查集）

题目链接

题面：

题意：
给定一张n个点m条边的无向图，每条边都有颜色，询问两个点之间一共有多少种不同的连接。
每种连接经过的边只能有一种颜色，两个连接所经过的边的颜色不同视为两个连接不同。
具体参考样例。

每种颜色对点建立一个并查集，询问时枚举颜色，若在当前颜色的并查集下x与y联通，那么就存在当前颜色下的连接。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#include<bitset>
#include<map>
#include<set>
#define ll long long
#define pr make_pair
#define pb push_back
#define ui unsigned int
#define lc (cnt<<1)
#define rc (cnt<<1|1)
#define tlen(x)  (t[(x)].r-t[(x)].l+1)
#define tmid (l+r)>>1
#define forhead for(int i=head[x];i;i=nt[i])
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)>(y)?(y):(x))
using namespace std;
const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const double dnf=1e18;
const int mod=1000000007;
const double eps=1e-8;
const double pi=acos(-1.0);
const int maxn=110;
const int maxm=100100;
const int up=100000;

int f[maxn][maxn];
int n,m;

int fi(int c,int x)
{
    if(f[c][x]!=x)
        f[c][x]=fi(c,f[c][x]);
    return f[c][x];
}

void init(void)
{
    for(int i=0;i<maxn;i++)
    {
        for(int j=0;j<maxn;j++)
            f[i][j]=j;
    }
}

void add(int x,int y,int c)
{
    int xx=fi(c,x);
    int yy=fi(c,y);
    if(xx==yy) return;
    f[c][xx]=f[c][yy];
}

int ask(int x,int y)
{
    int ans=0;
    for(int i=1;i<=m;i++)
    {
        if(fi(i,x)==fi(i,y)) ans++;
    }
    return ans;
}

int main(void)
{
    init();
    scanf("%d%d",&n,&m);
    int x,y,z;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
    }
    int q;
    scanf("%d",&q);
    for(int i=1;i<=q;i++)
    {
        scanf("%d%d",&x,&y);
        printf("%d\n",ask(x,y));
    }
    return 0;
}