http://acm.hdu.edu.cn/showproblem.php?pid=5418
g[i][j]:i 和 j 之间的最优解,通过 Floyd 算法求解
简洁的代码模板
const int inf = 0x3f;
const int maxn = 20;
int dp[1 << 20][maxn], mp[maxn][maxn], n;
int main() {
ios::sync_with_stdio(false);
cin >> n;
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
cin >> mp[i][j];
}
}
memset(dp, inf, sizeof(dp));
dp[1][0] = 0;
for (int i = 1; i < (1 << n); i += 2) {
for (int j = 0; j < n; j++) {
if ((i >> j) & 1) {
for (int k = 0; k < n; k++) {
if ((i - (1 << j)) >> k & 1) {
dp[i][j] = min(dp[i][j], dp[i - (1 << j)][k] + mp[k][j]);
}
}
}
}
}
cout << dp[(1 << n) - 1][n - 1] << endl;
return 0;
}
复杂代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=20,maxs=(1<<16)+5;
int n,m,g[maxn][maxn],c[maxs],dp[maxs][maxn];
//dp[s][i]:s 为曾到达过的城市的下标的映射,i是最后一个到达的城市的下标
int main(){
int t,cas=0,S,u,v,w;
for(int i=0;i<16;i++) c[1<<i]=i+1;
scanf("%d",&t);
while(t--){
memset(g,inf,sizeof(g));
memset(dp,inf,sizeof(dp));
scanf("%d%d",&n,&m);
S=(1<<n)-1;
for(int i=1;i<=n;i++) g[i][i]=0;
while(m--){
scanf("%d%d%d",&u,&v,&w);
g[u][v]=g[v][u]=min(g[v][u],w);
}
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<n;j++)
g[i][j]=min(g[i][j],g[i][k]+g[k][j]);
dp[1][1]=0;
//重点:3for
for(int s=1;s<S;s++){
for(int lst,i=s;i>0;i&=i-1){
lst=c[-i&i];//遍历最后一位1
if(dp[s][lst]==inf) continue; //没出现过,就跳过
//lst,nex分别为上次和这次的结尾;s,cs分别为上次和这次共出现的
//s和S^s关于S互补。所以下面这个for遍历剩余的未选的数
for(int cs,nex,d=S^s;d>0;d&=d-1){
nex=c[-d&d],cs=s|(-d&d);//也是按最后一位1的顺序遍历
dp[cs][nex]=min(dp[cs][nex],dp[s][lst]+g[lst][nex]);
}
}
}
//dp[S][i]:S表示每个城市都访问过,i表示最后经过哪个城市回到起点
for(int i=2;i<=n;i++)
dp[S][1]=min(dp[S][1],dp[S][i]+g[i][1]);
printf("%d\n",dp[S][1]);
}
}