Codeforces Round #286 (Div. 2) B. Mr. Kitayuta's Colorful Graph dfs

思路:因为color种类不超过100种,根据题目数据量,直接对每种color进行枚举,再进行dfs判断该种颜色下两点是否连通即可。

代码如下:

#include <cstdio>
#include <cstring>
using namespace std;
#define N 105
int head[N][N], idx;
struct Edge{
    int to, next;
}edges[N * 2];
bool vis[N];

void add(int from, int to, int color){
    ++idx;
    edges[idx].to = to, edges[idx].next = head[from][color], head[from][color] = idx;
}

bool is_connected(int u, int v, int color){
    vis[u] = true;
    for(int i = head[u][color]; i != 0; i = edges[i].next){
        int to = edges[i].to;
        if(to == v)
            return true;
        if(!vis[to] && is_connected(to, v, color))
            return true;
    }
    return false;
}

int main(){
    int n, m, q;
    scanf("%d %d", &n, &m);
    for(int i = 0; i < m; ++i){
        int v1, v2, color;
        scanf("%d %d %d", &v1, &v2, &color);
        add(v1, v2, color);
        add(v2, v1, color);
    }
    scanf("%d", &q);
    for(int i = 0; i < q; ++i){
        int u, v, ans = 0;
        scanf("%d %d", &u, &v);
        for(int j = 1; j <= m; ++j){
            memset(vis, false, sizeof(vis));
            if(is_connected(u, v, j))
                ans ++;
        }
        printf("%d\n", ans);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值