首页 > 代码库 > [POJ 3311]Hie with the Pie——再谈TSP问题的DP解法
[POJ 3311]Hie with the Pie——再谈TSP问题的DP解法
题目连接:
http://poj.org/problem?id=3311
题目大意:有n+1个点,给出点0~n的每两个点之间的距离,求这个图上TSP问题的最小解
思路:用二进制数来表示访问过的城市集合,f[{S}][j]=已经访问过的城市集合为S,访问了j个城市,所需的最少花费。
这里提一下二进制数表示集合的方法(这里不妨设集合中最多有n个元素):
如果集合S中最多会出现n个元素,则用长度为n的二进制数来表示集合S,每一位代表一个元素,该位为0表示该元素在集合S中不存在,为1表示该元素在集合S中存在
位数 4 3 2 1
S 1 0 1 1
这个集合S里有元素1、2、4
下面是二进制数表示几种集合运算的方法
1、集合S的全集U=(1<<n)-1
2、检查集合S中是否含元素i S&(1<<(i-1)) (返回0表示不存在,返回1表示存在)
3、从集合S中去除元素i S^(1<<(i-1))
下面是本题的思路:
首先对整个图跑一次Floyd多源最短路,得到两两点之间的最短距离,然后用DP求解,f[{S}][j]=已经访问过的城市集合为S,访问了j个城市,所需的最少花费。f[S][i]=min{f[S-{j}][j]+dist[j][i]}
最后得到的答案ans=min(f[全集][i]+dist[i][0])
代码:
#include <iostream> #include <stdio.h> #include <string.h> #include <stdlib.h> #include <algorithm> #define MAXN 15 #define MAXM 1<<15 #define INF 0x3f3f3f3f using namespace std; int f[MAXM][MAXN]; //f[{S}][j]=已经访问过的城市集合为S,访问了j个城市,所需的最少花费 int dist[MAXN][MAXN]; //点与点之间的距离 int n; int min(int a,int b) { if(a<b) return a; return b; } void Floyd() { for(int k=0;k<=n;k++) for(int i=0;i<=n;i++) for(int j=0;j<=n;j++) dist[i][j]=min(dist[i][j],dist[i][k]+dist[k][j]); } int TSP() //DP求TSP { memset(f,0x7f,sizeof(f)); for(int s=0;s<(1<<n);s++) //枚举访问城市集合S,全集为(1<<n)-1 for(int i=1;i<=n;i++) //枚举最近访问过的城市i if(s&(1<<(i-1))) //city(i)∈S { if(s==(1<<(i-1))) //{city(i)}==S f[s][i]=dist[0][i]; else { for(int j=1;j<=n;j++) //枚举上一次访问的城市j if((s&(1<<(j-1)))&&i!=j) //城市j不和i相同 f[s][i]=min(f[s][i],f[s^(1<<(i-1))][j]+dist[j][i]); //Cs {city(J)}=s^(1<<(i-1)) } } int ans=INF; for(int i=1;i<=n;i++) ans=min(ans,f[(1<<n)-1][i]+dist[i][0]); return ans; } int main() { while(scanf("%d",&n)&&n) { memset(dist,0,sizeof(dist)); for(int i=0;i<=n;i++) for(int j=0;j<=n;j++) scanf("%d",&dist[i][j]); Floyd(); printf("%d\n",TSP()); } return 0; }
[POJ 3311]Hie with the Pie——再谈TSP问题的DP解法
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。