蓝桥杯——逗志芃的暴走 (C++)

题目来源：蓝桥杯算法训练

知识点：搜索、Floyd算法

题目分析

本题可以看成是图问题中寻找最短路径的变体，只是在这里一个顶点可以经过多次。这里的图是一个完全图，数组a[i][j]表示从地点i到地点j需要的时间。

可以发现，直接从地点i到地点j可能时间较长，而先从地点i到地点k，再从地点k到地点j能获得更短的时间。因此，在 DFS 搜索前先使用弗洛伊德算法对n*n的矩阵进行处理，获得元素a[i][j]的最小值。然后在图上进行 DFS 搜索即可。

关于弗洛伊德算法可以自行查找，本质就是本代码中preprocess()函数的意思。

代码

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

const int MAX = 35;
int tim[MAX][MAX];
int des[MAX];
int vis[MAX];
int n, m, num; 
int ans;

void preprocess() {
	for(int k=1; k<=n; ++k){
		for(int i=1; i<=n; ++i){
			for(int j=1; j<=n; ++j){
				tim[i][j] = min(tim[i][j],tim[i][k]+tim[k][j]);
			}
		}
	}
}

void dfs(int cur, int cnt, int t) {
	if(cnt == num) {
		ans = min(ans, t);
		return;
	}
	for(int i=1; i<=n; i++) {
		if(des[i] && !vis[i] && tim[cur][i]+t <= ans) {
			vis[i] = 1;
			dfs(i, cnt+1, t+tim[cur][i]);
			vis[i] = 0;
		}
	}
}

int main() {
	cin >> n;
	for(int i=1; i<=n; i++) {
		for(int j=1; j<=n; j++) {
			cin >> tim[i][j];
		}
	}
	
	preprocess();

	cin >> m;	
	int a;
	num = 0;
	memset(des, 0, sizeof(des));
	
	for(int i=1; i<=m; i++) {
		cin >> a;
		if(!des[a]) {
			num++;
			des[a] = 1;
		}
	}
	
	ans = 1e6;
	
	for(int i=1; i<=n; i++) {
		if(des[i]) {
			memset(vis, 0, sizeof(vis));
			vis[i] = 1;
			dfs(i, 1, 0);
		}
	}
	
	cout << ans << endl;
	
	return 0;
}