Codeforces Round #286 (Div. 2), problem: (B) Mr. Kitayuta's Colorful Graph 【弗洛伊德floyd】

题意

首行给出N和M，代表有N个点。接下来有M行，每行三个数字A,B,C，代表A到B有颜色为C的绳（也可以是线，随便是什么吧）。接着给出Q，代表Q次询问。接下来Q行，每行给出A和B，只有相同颜色的绳才可以把AB联通，问联通AB的绳子一共有多少种

思路

求两点之间有多少不同颜色的路径

范围比较小 可以直接floyd

code

#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
int n,m,q;
const int maxn=200;
int dp[maxn][maxn][maxn];
int main(){
    cin>>n>>m;
    for(int i=0;i<m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        dp[c][a][b]=dp[c][b][a]=1;
    }
    for(int c=1;c<=m;c++)
    for(int k=1;k<=n;k++)
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++){
        if(!dp[c][i][j])
            dp[c][i][j]=dp[c][j][i]=(dp[c][i][k]&&dp[c][k][j]);
    }
    cin>>q;
    while(q--){
        int a,b,ans=0;
        cin>>a>>b;
        for(int c=1;c<=m;c++)
            if(dp[c][a][b]) ++ans;
        cout<<ans<<endl;
    }
    return 0;
}

学如逆水行舟，不进则退