东华OJ 进阶题31 文化之旅

问题描述 :

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

样例输入

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

样例输出

10

输入输出样例说明

　　路线为1 -> 2。

输入说明 :

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

输入范例 :

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

输出范例 :

10

完整C++代码

#include <bits/stdc++.h>
using namespace std;

int N, K, M, S, T, res = -1;
// 国家地图、文化地图、各个国家的文化、是否到达、是否学习
int edge[110][110], culture[110][110], country[110], visit[110], learn[110];

// 判断学过的文化是否与x国家有冲突
int check(int x, int y)
{
	if (culture[country[y]][country[x]]) return 0;
	if (culture[country[T]][country[y]]) return 0;
	for (int i = 1; i <= learn[0]; i++) {
		if (culture[learn[y]][country[i]]) return 0;
	}
	return 1;
}

void dfs(int deep, int dis)
{
	if (res != -1 && dis >= res) return;
	if (deep == T) {
		res = dis;
		return;
	}
	for (int i = 1; i <= N; i++) {
		if (deep == i) continue;
		if (visit[i] == 0 && check(deep, i)) {
			visit[i] = 1;
			learn[0]++;
			learn[learn[0]] = country[i];
			dfs(i, dis + edge[deep][i]);
			learn[0]--;
			visit[i] = 0;
		}
	}
}

int main()
{
	cin >> N >> K >> M >> S >> T;
	for (int i = 1; i <= N; i++) {
		cin >> country[i];
	}
	for (int i = 1; i <= K; i++) {
		for (int j = 1; j <= K; j++) {
			cin >> culture[i][j];
		}
	}
	// 默认相互不可到达 
	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N; j++) {
			edge[i][j] = 0x3f3f3f;
		}
	}
	int u, v, d;
	for (int i = 1; i <= M; i++) {
		cin >> u >> v >> d;
		// 这个顺序别搞反了，应该是你的国家不排斥我 我就可以过去，不管我是否排斥你
		if (!culture[country[v]][country[u]] && country[u] != country[v]) edge[u][v] = min(edge[u][v], d);
		if (!culture[country[u]][country[v]] && country[u] != country[v]) edge[v][u] = min(edge[v][u], d);
	}
	// 存储已经学过的 
	learn[0] = 1;
	learn[1] = country[S];
	visit[S] = 1;
	dfs(S, 0);
	cout << res << endl;
	return 0;
}

😋欢迎大伙私信或者评论区交流讨论😋