java算法 蓝桥杯 文化之旅

问题描述

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

输入格式

　　第一行为五个整数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）。

样例输入

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

样例输出

-1

输入输出样例说明

　　由于到国家2必须要经过国家1，而国家2的文明却排斥国家1的文明，所以不可能到达国家2。

样例输入

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

样例输出

输入输出样例说明

　　路线为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≤ N^2，1≤ki≤K，1≤u, v≤N，1≤d≤1000，S≠T，1 ≤S, T≤N。

 

package 蓝桥杯java算法训练;

import java.util.Scanner;

public class _6文化之旅 {
   public static void main(String[] args) {
	Scanner scanner = new Scanner(System.in);
	int n,k,m,s,t,u,v,d;
	int wuqiong = 99999999;
	n = scanner.nextInt();//n个国家
	k = scanner.nextInt();//k种文化
	m = scanner.nextInt();//m条道路
	s = scanner.nextInt();//起点s
	t = scanner.nextInt();//终点t

	int[] c = new int[n+1];//c[i]国家i的文化是C[i]
	int[][] a = new int[k+1][k+1];//第i行第j个数是a(i,j)=1排斥，等于0不排斥
	int[][] dis = new int[2000][2000];//记录路径距离
	for (int i = 1; i <= n; i++) {
		c[i] = scanner.nextInt();
	}//输入c[]
	for (int i = 1; i <= k; i++) {
		for (int j = 1; j <= k; j++) {
			a[i][j] = scanner.nextInt();
		}
	}//输入a[][]
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++) {
			if (i != j) {
				dis[i][j] = wuqiong;
			}
		}
	}//初始化每个国家之间都没有路径
	for (int i = 1; i <= m; i++) {
		u = scanner.nextInt();
		v = scanner.nextInt();
		d = scanner.nextInt();
		//u v两个国家之间有道路，看文化是否排斥，要是不排斥，找最小的距离
		if(a[c[v]][c[u]] == 0) dis[u][v]=d<dis[u][v]?d:dis[u][v];
        if(a[c[u]][c[v]] == 0) dis[v][u]=d<dis[v][u]?d:dis[v][u];
	}//输入uvd
	 for(int k1=1;k1<=n;k1++)
	        for(int i=1;i<=n;i++)
	            for(int j=1;j<=n;j++)
	                if(dis[i][j]>dis[i][k1]+dis[k1][j])
	                    if(a[c[j]][c[i]]==0&&a[c[k1]][c[j]]==0&&a[c[i]][c[k1]]==0)
	                        dis[i][j]=dis[i][k1]+dis[k1][j];
	    if(dis[s][t]>=wuqiong)
	       System.out.println("-1");
	    else
	    	System.out.println(dis[s][t]);

}
}