Problem:
有n个城市,John从第一个城市出发,最终回到第一个城市,每个城市只经过一次,求最小代价?
Solution:
先利用floyd求出两两城市之间的最短路,然后以1为起点跑一下TSP算法,min(dp[1<<(n-1)][k]+dist[k][1])就是最小代价。
TSP算法解释:
定义dp[i][k]为集合i中最后一个到达k节点所付出的代价。i的每一个二进制位都代表对应的元素是否存在。状态转换为:
dp[i|(1<<(j-1))][j] = min(dp[i|(1<<(j-1))][j], dp[i][k]+G[k][j]);
可以理解为不包含j元素的集合i中最后一个元素为k,加上k到j的距离就是包含j元素的集合中最后一个元素是j的值,遍历所有的k找到最小值即可。dp[1][1] = 0开始遍历则说明了起点是1,如果dp[1<<(k-1)][k] = 0,则说明k可以使起点,表示集合中只有k这个元素且重点为k。
#include<cstdio>
#include<iostream>
#include<sstream>
#include<cstdlib>
#include<cmath>
#include<cctype>
#include<string>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<ctime>
#include<vector>
#include<fstream>
#include<list>
#include<numeric>
#include<functional>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define ms(s) memset(s,0,sizeof(s))
const double PI = 3.141592653589;
const int INF = 0x3fffffff;
int dp[1<<17][17];
int G[17][17];
int n;
int main() {
// freopen("/Users/really/Documents/code/input","r",stdin);
// freopen("/Users/really/Documents/code/output","w",stdout);
ios::sync_with_stdio(false);
int t, m, u, v, w;
cin >> t;
while(t--) {
cin >> n >> m;
//init
for(int i = 0; i <= n; i++)
for(int j = 0; j <= n; j++)
G[i][j] = (i==j) ? 0 : INF;
for(int i = 0; i <= (1<<n); i++)
for(int j = 0; j <= n; j++)
dp[i][j] = INF;
//input
for(int i = 0; i < m; i++) {
cin >> u >> v >> w;
G[u][v] = G[v][u] = min(G[u][v], w);
}
//Floyd
for(int k = 1; k <= n; k++)
for(int i = 1; i <= n; i++)
for(int j = 1; j <= n; j++)
G[i][j] = G[j][i] = min(G[i][j], G[i][k]+G[k][j]);
//dp
dp[1][1] = 0;
for(int i = 1; i < (1<<n); i++) {
for(int j = 1; j <= n; j++) {
if((i & (1<<(j-1))) == 0) {
for(int k = 1; k <= n; k++) {
if(i & (1<<(k-1)))
dp[i|(1<<(j-1))][j] = min(dp[i|(1<<(j-1))][j], dp[i][k]+G[k][j]);
}
}
}
}
//output
int ans = dp[(1<<n)-1][n] + G[1][n];
for(int i = 1; i <= n; i++)
ans = min(ans, dp[(1<<n)-1][i] + G[1][i]);
cout << ans << endl;
}
return 0;
}