题目连接:http://codeforces.com/contest/505/problem/B
觉得很奇怪在第二题就出现了图的问题、不过这题目并不难,有很多方法可以解决。
一开始就想到了DFS,感觉时间会超时,又想到并查集。。。。这个好像不太靠谱,最后决定用BFS写,找出同种颜色的路径下相互连通的点集合,打成表按照要求直接输出。可是WA在第6组数据。
赛后写了一下DFS 竟然AC了 而且时间是46MS
DFS思路就是存储所有的边放入Map数组,然后对100种颜色进行DFS,最后想想数据比较小 N M都小于100,,然后就没有然后了
对效率的估计还有待加强- -总是把时间估计多了
#include<bits/stdc++.h>
#define for(i,limit) for(int i = 0 ; i < limit ; i++)
using namespace std;
int n , m , q;
bool Map[105][105][105] , vis[105];
void dfs(int v , int color)
{
vis[v] = true;
for(i,100)
{
if(Map[v][i][color] == true && vis[i] == false)
dfs(i,color);
}
}
int main()
{
int v1 , v2 , color;
cin >> n >> m;
for(i,m)
{
cin >> v1 >> v2 >> color;
Map[v1][v2][color] = Map[v2][v1][color] = true;
}
cin >> q;
for(i,q)
{
int cnt = 0;
cin >> v1 >> v2;
for(color,100)
{
memset(vis , false , sizeof(vis));
dfs(v1,color);
if(vis[v2] == true)
cnt++;
}
cout << cnt << endl;
}
return 0;
}