这个是经典状态压缩dp tsp问题的变形
首先来看看《挑战程序设计竞赛》讲解tsp问题:
给定一个n个定顶点组成的带权有向图的距离矩阵d(i,j)(INF表示没有变)。要求从顶点0出发,经过每个顶点恰好一次后再回到顶点0。问所经过得边的总权重的最小值是多少?
限制条件
1<n<16
0<d(i,j)<1000
tsp问题是np困难的。没有已知的的多项式时间的高效算法可以解决这一问题。
所有可能的路线共有(n-1)!种。这是个非常大的值。即使在本题中n已经很小了。仍然无法试遍每一种情况。对于这个问题我们可以使用dp来解决。首先让我们试着写出它的递推式。
假设现在已经访问过得顶点的集合(起点0当做还未访问过的顶点)为S.当前所在顶点为v,用dp[S][v]表示从v出发访问剩余的所有顶点。最终回到顶点0的路径的权重总和的最小值。由于从v出发可以移动到任意一个节点u不属于S,因此有如下递推式:
dp[V][0]=0;
dp[S][v]=min{dp[S∪{u}][u]+d(v,u)|u不属于S};
我只要按照这个递推式进行计算就可以了。这样就可以在O(2^n*n^2)的时间里完成计算。
有n个城市,一个人去旅游,他想游遍n个城市,且每个城市最多去两次,问最短的路程是多少。
起点任意,每两个城市之间可能会有多条路,也可能会没有路。如果能游遍所有的城市
输出最少的,不能的话输出-1这道题与经典tsp问题用的是同一种dp递推式。区别是:(1)不需要回路(2)顶点限制的次数+1
经典tsp问题它要求顶点只能遍历一次,所以对于dp的状态,记录的是该顶点的入度。而在这道题里入度的要求则变成了indegree小于等于2.所以问题就从经典tsp的二进制的状态压缩变成了三进制的状态压缩.
对于三进制的状态转移因为不能用位运算了,所以就要对三进制的操作进行预处理并用数组保存。
#include <stdio.h> #include <string.h> #include <stdlib.h> #define Max_V 12 #define INF 0x3f3f3f3f int n,m; int g[Max_V][Max_V]; int ter[60000][Max_V];//ter[S][v]表示在S状态中v顶点的入度 int bit[]={1,3,9,27,81,243,729,2187,6561,19683,59049}; int dp[60000][Max_V]; int findMin(int x,int y) { return x>y?y:x; } void ternary_init() {//对三进制的操作做预处理 <span style="white-space:pre"> </span>int i, j, b; for(i=0;i<bit[10];i++) { b=i; for(j=0;j<10;j++) { ter[i][j] =b % 3; b/=3; } } } void init() {//初始化 memset(g,-1,sizeof(g)); memset(dp,INF,sizeof(dp)); for(int i=0;i<n;i++) dp[bit[i]][i]=0; } int main() { int i; int u,v,w; ternary_init(); while(~scanf("%d%d",&n,&m)) { init(); for(i=0;i<m;i++) { scanf("%d%d%d",&u,&v,&w); if(g[u-1][v-1]>w||g[u-1][v-1]==-1) g[u-1][v-1]=g[v-1][u-1]=w; } int ans=INF; for(int S=0;S<bit[n];S++) {//遍历所有状态 int flag=true; for(u=0;u<n;u++) { if(ter[S][u]==0) flag=false; if(dp[S][u]==INF) continue; for(v=0;v<n;v++) { if(u==v||g[u][v]==-1||ter[S][v]>=2) continue; dp[S+bit[v]][v]=findMin(dp[S+bit[v]][v],dp[S][u]+g[u][v]);//状态转移 } } if(flag) {//所有顶点的入度都大于0 for(u=0;u<n;u++) ans=findMin(ans,dp[S][u]); } } if(ans==INF) printf("-1\n"); else printf("%d\n",ans); } return 0; }