问题 D: 文化之旅

时间限制: 1 Sec 内存限制: 128 MB

题目描述

有一位使者要游历各国，他每到一个国家，都能学到一种文化，但他不愿意学习任何一种文化超过一次（即如果他学习了某种文化，则他就不能到达其他有这种文化的国家）。不同的国家可能有相同的文化。不同文化的国家对其他文化的看法不同，有些文化会排斥外来文化（即如果他学习了某种文化，则他不能到达排斥这种文化的其他国家）。
现给定各个国家间的地理关系，各个国家的文化，每种文化对其他文化的看法，以及这位使者游历的起点和终点（在起点和终点也会学习当地的文化），国家间的道路距离，试求从起点到终点最少需走多少路。

输入

第一行为五个整数 N，K，M，S，T，每两个整数之间用一个空格隔开，依次代表国家个数（国家编号为 1 到 N），文化种数（文化编号为 1 到 K），道路的条数，以及起点和终点的编号（保证 S 不等于 T）；
第二行为 N 个整数，每两个整数之间用一个空格隔开，其中第 i 个数 Ci，表示国家 i的文化为 Ci。
接下来的 K 行，每行 K 个整数，每两个整数之间用一个空格隔开，记第 i 行的第 j 个数为 aij，aij= 1 表示文化 i 排斥外来文化 j（i 等于 j 时表示排斥相同文化的外来人），aij= 0 表示不排斥（注意 i 排斥 j 并不保证 j 一定也排斥 i）。
接下来的 M 行，每行三个整数 u，v，d，每两个整数之间用一个空格隔开，表示国家 u与国家 v 有一条距离为 d 的可双向通行的道路（保证 u 不等于 v，两个国家之间可能有多条道路）。
输出
输出只有一行，一个整数，表示使者从起点国家到达终点国家最少需要走的距离数（如果无解则输出-1）。

样例输入 Copy

2 2 1 1 2
1 2
0 1
1 0
1 2 10

样例输出 Copy

-1

提示

对于100%的数据 有2≤N≤100，1≤K≤100，1≤M≤N^2，1≤ki≤K，1≤u，v≤N，1≤d≤1000，S≠T，1≤S,T≤N。

flody算法求最短路，n<=100范围很小，暴力即可
代码如下

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int N=110;
int n,k,m,s,t;
int c[N],a[N][N],mp[N][N];
int main()
{ 
    cin>>n>>k>>m>>s>>t;
    for(int i = 1; i<=n; i++) //国家文化
    { 
        cin>>c[i];
    }
    for(int i = 1; i<=k; i++)
        for(int j = 1; j<=k; j++)
            cin>>a[i][j];
    for(int i = 1; i<=n; i++)
        for(int j = 1; j<=n; j++)
        { 
            if(i!=j)
                mp[i][j]=99999999;
        }
    while(m--)
    { 
        int u,v,d;
        cin>>u>>v>>d;
        if(!a[c[u]][c[v]])mp[u][v]=min(mp[u][v],d);
        if(!a[c[v]][c[u]])mp[v][u]=min(mp[v][u],d);
    }
    for(int k = 1; k<=n; k++)//flody,k为中间国，用来松弛
        for(int i = 1; i<=n; i++)
            for(int j = 1; j<=n; j++)
            { 
                if(!a[c[i]][c[k]]&&!a[c[i]][c[j]]&&!a[c[k]][c[j]])
                { 
                    mp[i][j]=min(mp[i][j],mp[i][k]+mp[k][j]);
                }
            }
    if(mp[s][t]==99999999)cout<<-1;
    else cout<<mp[s][t];
    return 0;
}