问题链接: AcWing 91. 最短Hamilton路径
问题描述
分析
如果用暴力枚举做法来做的话,需要枚举(n-2)!*20——第一个点和最后一个点固定(n-2)!种状态,每次枚举一种状态要计算其路径长度最多计算20次,
所以时间复杂度是O(n*n!)
这里考虑用状态压缩DP来做,状态压缩DP的核心思想是用位运算来表示状态
状态表示:
f[i][j]表示当前选的点为i,且以j为结尾的最短路径
例如,f[7][2],5的二进制为111这表示选择了0、1、2号点,且以2号点为结尾的最短路径
状态转移:
这里的状态转移考虑从上一个点转移过来,例如f[7][2]的终点为2,可以从0、1号点转移过来
(状态转移时要去掉2号点),看代码更加容易理解一些
代码如下:
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
const int N=20,M=1<<20;
int f[M][N];
int w[N][N];
int main(){
int n;
cin>>n;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++) scanf("%d",&w[i][j]);
memset(f,0x3f,sizeof f);
f[1][0]=0;//初始状态0号点到0号点的距离为0,0的状态表示为1(第0位为1)
for(int i=0;i<1<<n;i++)
for(int j=0;j<n;j++)
if((i>>j)&1)//当前以j为结尾,必须包含j点
for(int k=0;k<n;k++)//遍历j点前一个点
if((i>>k)&1) f[i][j]=min(f[i][j],f[i-(1<<j)][k]+w[k][j]);
cout<<f[(1<<n)-1][n-1];//最终点为n-1,其余点都走过了用(1<<n)-1表示
return 0;
}
//注意+-运算优先级高于位运算