二维并查集——CodeForces - 505B Mr. Kitayuta's Colorful Graph

原题链接CodeForces - 505B Mr. Kitayuta’s Colorful Graph

一个无向图包含n个点m条边，顶点编号从1到n。 对于每条边有颜色ci， 连接着顶点 ai 和 b i.

下面有q个询问.

每条询问有两个整数 — ui 和 vi.

找到满足下面条件的颜色个数: 同一种颜色的路径连接顶点 ui 和 顶点 vi

Input

第一行是两个整数 — n and m (2 ≤ n ≤ 100, 1 ≤ m ≤ 100), 代表着定点个数和边的个数

接下来m行有三个整数 — ai, bi (1 ≤ ai < bi ≤ n) and ci (1 ≤ ci ≤ m).

下一行有一个整数 — q (1 ≤ q ≤ 100), 代表询问次数

接下来q行，每行两个整数 — ui 和 vi (1 ≤ ui, vi ≤ n). 默认 ui ≠ vi.

Output

对于每次询问，在单独的一行输出答案

Example

Input

4 5

1 2 1

1 2 2

2 3 1

2 3 3

2 4 3

3

1 2

3 4

1 4

Output

2

1

0

Input

5 7

1 5 1

2 5 1

3 5 1

4 5 1

1 2 2

2 3 2

3 4 2

5

1 5

5 1

2 5

1 5

1 4

Output

1

1

1

1

2

思路：数据量很小 可以用搜索

并查集解决连接问题 每种颜色都需要开一个并查集 所以是二维的

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <stack>
#include <vector>
#define max_ 110
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;
int pre[max_][max_];
int n,m;
int find(int x,int t)//t代表颜色种类
{
return pre[t][x]==x?x:find(pre[t][x],t);
}
void join(int x,int y,int t)//t代表颜色种类
{
x=find(x,t);
y=find(y,t);
if(x!=y)
pre[t][x]=y;
}
int main(int argc, char const *argv[])
{
scanf("%d%d",&n,&m);
int i,j,r=m;
for(i=1;i<max_;i++)   //初始化
for(j=1;j<max_;j++)
pre[i][j]=j;
while(m--)
{
int x,y,t;
scanf("%d%d%d",&x,&y,&t);//t代表颜色种类
join(x,y,t);
}
int q;
scanf("%d",&q);
while(q--)
{
int u,v;
scanf("%d%d",&u,&v);
int cnt=0;
for(i=1;i<=r;i++)
{
if(find(u,i)==find(v,i))
cnt++;
}
printf("%d\n",cnt);
}
return 0;
}