有一位使者要游历各国,他每到一个国家,都能学到一种文化,但他不愿意学习任何一种文化超过一次(即如果他学习了某种文化,则他就不能到达其他有这种文化的国家)。不同的国家可能有相同的文化。不同文化的国家对其他文化的看法不同,有些文化会排斥外来文化(即如果他学习了某种文化,则他不能到达排斥这种文化的其他国家)。
现给定各个国家间的地理关系,各个国家的文化,每种文化对其他文化的看法,以及这位使者游历的起点和终点(在起点和终点也会学习当地的文化),国家间的道路距离,试求从起点到终点最少需走多少路。
第一行为五个整数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
输入样例2
2 2 1 1 2
1 2
0 1
0 0
1 2 10
输出样例1
-1
输出样例2
10
【输入输出样例1说明】
由于到国家2必须要经过国家1,而国家2的文明却排斥国家1的文明,所以不可能到达国家2。
【输入输出样例2说明】
路线为1 -> 2。
【数据范围】
对于20%的数据,有2≤N≤8,K≤5;
对于30%的数据,有2≤N≤10,K≤5;
对于50%的数据,有2≤N≤20,K≤8;
对于70%的数据,有2≤N≤100,K≤10;
对于100%的数据,有2≤N≤100,1≤K≤100,1≤M≤N2,1≤ki≤K,1≤u,v≤N,1≤d≤1000,S≠T,1 ≤S, T≤N。
思路:dfs
代码实现:
1 #include<cstdio> 2 int n,k,m,s,t,ans=-1; 3 int a,b,c; 4 int v[110]; 5 bool u[110],map[110][110]; 6 int h[110],hs=1; 7 struct edge{int s,n,d;}e[20010]; 8 void dfs(int n,int s){ 9 if(n==t){if(s<ans||ans==-1) ans=s;return;} 10 if(s>ans&&ans!=-1) return; 11 for(int i=h[n];i;i=e[i].n) 12 if(!u[v[e[i].s]]){ 13 for(int j=1;j<=k;j++) if(map[v[e[i].s]][j]&&u[j]) a=-1; 14 if(a<0){a=0;continue;} 15 u[v[e[i].s]]=1; 16 dfs(e[i].s,s+e[i].d); 17 u[v[e[i].s]]=0; 18 } 19 } 20 int main(){ 21 scanf("%d%d%d%d%d",&n,&k,&m,&s,&t); 22 for(int i=1;i<=n;i++) scanf("%d",&v[i]); 23 for(int i=1;i<=k;i++) for(int j=1;j<=k;j++) scanf("%d",&map[i][j]); 24 for(int i=1;i<=m;i++){ 25 scanf("%d%d%d",&a,&b,&c); 26 e[++hs]=(edge){b,h[a],c},h[a]=hs; 27 e[++hs]=(edge){a,h[b],c},h[b]=hs; 28 } 29 u[v[s]]=1; 30 dfs(s,0); 31 printf("%d",ans); 32 return 0; 33 }
我连最简单的剪枝都忘了。
题目来源:CODE[VS],洛谷