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

博客讨论了如何使用二维并查集解决CodeForces上的505B问题,即在一个无向图中处理颜色相关的查询。题目要求找出两个给定点之间通过直接或间接的特定颜色边连接的数量。输入包括图的顶点数、边数、每条边的颜色和查询次数,输出是每次查询的答案。由于数据规模较小,可以采用并查集的数据结构来有效处理各个颜色的连通性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目链接:http://codeforces.com/problemset/problem/505/B点击打开链接

B. Mr. Kitayuta's Colorful Graph
time limit per test
1 second
memory limit per test
256 megabytes
input
standard input
output
standard output

Mr. Kitayuta has just bought an undirected graph consisting of n vertices and m edges. The vertices of the graph are numbered from 1 to n. Each edge, namely edge i, has a color ci, connecting vertex ai and bi.

Mr. Kitayuta wants you to process the following q queries.

In the i-th query, he gives you two integers — ui and vi.

Find the number of the colors that satisfy the following condition: the edges of that color connect vertex ui and vertex vi directly or indirectly.

Input

The first line of the input contains space-separated two integers — n and m (2 ≤ n ≤ 100, 1 ≤ m ≤ 100), denoting the number of the vertices and the number of the edges, respectively.

The next m lines contain space-separated three integers — aibi (1 ≤ ai < bi ≤ n) and ci (1 ≤ ci ≤ m). Note that there can be multiple edges between two vertices. However, there are no multiple edges of the same color between two vertices, that is, if i ≠ j(ai, bi, ci) ≠ (aj, bj, cj).

The next line contains a integer — q (1 ≤ q ≤ 100), denoting the number of the queries.

Then follows q lines, containing space-separated two integers — ui and vi (1 ≤ ui, vi ≤ n). It is guaranteed that ui ≠ vi.

Output

For each query, print the answer in a separate line.

Examples
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
Note

Let's consider the first sample. 

The figure above shows the first sample. 
  • Vertex 1 and vertex 2 are connected by color 1 and 2
  • Vertex 3 and vertex 4 are connected by color 3
  • Vertex 1 and vertex 4 are not connected by any single color. 


数据量很小  可以用并查集 主要就是构造每个颜色的并查集 这样才能区别每个地点是否与上一个地点相连

#include <bits/stdc++.h>
using namespace std;
#define maxn 200
int pre[maxn][maxn];
int findx(int col,int x)
{
    int r=x;
    while(pre[col][r]!=r)
    {
        r=pre[col][r];
    }
    int i=x;int j;
    while(pre[col][i]!=r)
    {
        j=pre[col][i];
        pre[col][i]=r;
        i=j;
    }
    return r;
}

void join (int col,int x,int y)
{
    int p1=findx(col,x);
    int p2=findx(col,y);
    if(p1!=p2)
    {
        pre[col][p2]=p1;
    }
}
int judge(int col,int x,int y)
{
    if(findx(col,x)!=findx(col,y))
        return 0;
    else
        return 1;
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=0;i<maxn;i++)
        for(int j=0;j<maxn;j++)
            pre[i][j]=j;
    for(int i=0;i<m;i++)
    {
        int mid1,mid2,mid3;
        scanf("%d%d%d",&mid1,&mid2,&mid3);
        join(mid3,mid1,mid2);
    }
    int t=0;
    scanf("%d",&t);
    for(int i=0;i<t;i++)
    {
        int mid1,mid2;
        scanf("%d%d",&mid1,&mid2);
        int ans=0;
        for(int i=1;i<=m;i++)
            ans+=judge(i,mid1,mid2);
        printf("%d\n",ans);
    }
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值