链接:https://ac.nowcoder.com/acm/problem/16577
来源:牛客网
题目描述
有一位使者要游历各国,他每到一个国家,都能学到一种文化,但他不愿意学习任何一种文化超过一次(即如果他学习了某种文化,则他就不能到达其他有这种文化的国家)。不同的国家可能有相同的文化。不同文化的国家对其他文化的看法不同,有些文化会排斥外来文化(即如果他学习了某种文化,则他不能到达排斥这种文化的其他国家)。
现给定各个国家间的地理关系,各个国家的文化,每种文化对其他文化的看法,以及这位使者游历的起点和终点(在起点和终点也会学习当地的文化),国家间的道路距离,试求从起点到终点最少需走多少路。
输入描述:
第一行为五个整数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)。
示例1
输入
复制
2 2 1 1 2
1 2
0 1
1 0
1 2 10
输出
复制
-1
说明
由于到国家2必须要经过国家1,而国家2的文明却排斥国家1的文明,所以不可能到达国家2。
示例2
输入
复制
2 2 1 1 2
1 2
0 1
0 0
1 2 10
输出
复制
10
说明
路线为1->2
解题思路:
解法1:dfs()
- 用vis[][]来存储彼此国家排斥的文化,visit[][]标记该国家的文化是否已经被访问过。
2)根据vis[i][c[s]]=1可判断标记visit[i]=1; 表明i这些文化的国家不可能被访问
3)通过dfs()在满足题目要求的基础上,找寻最短路。
c++ 代码
#include <bits/stdc++.h>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
using namespace std;
int n,k,m,s,t,ans=100000000,cnt=0;
int c[200];
int vis[200][200];
int visit[200];
struct node
{
int to,next,w;
}e