问题描述
假设有一个旅行商人要拜访n个城市,他必须选择所要走的路径,路径的限制是每个城市只能拜访一次,而且最后要回到原来出发的城市。路径的选择目标是要求得的路径路程为所有路径之中的最小值。
动态规划
假设dp(i,V)表示从原点(0)出发经过集合V中的各个顶点各一次回到i点时的最短路径距离(V中不包含i)
则dp(i,V)=min(dist(i,k)+dp(k,V-{k}))(k∈V)
dp(i,0)=dist(i,0)
V的表示:以二进制来表示,例如0b110表示点3和点2在集合V中
源代码
/**
@动态规划-旅行商(TSP)问题
*/
#include <iostream>
#include <vector>
#include <math.h>
using namespace std;
#define INF 9999
void main()
{
int n;
//输入城市个数
cin>>n;
vector<vector<int>> dist(n,vector<int>(n));//城市与城市之间的距离
int i,j;
//输入城市与城市之间的距离
for(i=0;i<n;i++)
for(j=0;j<n;j++)
cin>>dist[i][j];
//以二进制数来表示V中所包含的城市
int t=1<<(n-1);
vector<vector<int>> dp(n,vector<int>(t));//dp[i][j]表示从0到i经过j中的点的最短距离
vector<vector<int>> path(n,vector<int>(t));//记录最短路径
for(i=0;i<n;i++)
for(j=0;j<t;j++)
dp[i][j]=INF;
//初始化dp第0列
for(i=0;i<n;i++)
dp[i][0]=dist[i][0];
for(i=1;i<t;i++)
{
for(j=1;j<n;j++)
{
if(((1<<(j-1))&i)==0)//V中不包含j
{
for(int k=1;k<n;k++)
{
int temp=dp[j][i];
if((1<<(k-1))&i)//V中包含k
dp[j][i]=min(dp[j][i],dist[j][k]+dp[k][i-(1<<(k-1))]);
if(temp!=dp[j][i])//dp[j][i]的值更新,更新路径
path[j][i]=k;
}
}
}
}
for(int k=1;k<n;k++)
{
int temp=dp[0][t-1];
if((1<<(k-1))&(t-1))
dp[0][t-1]=min(dp[0][t-1],dist[0][k]+dp[k][t-1-(1<<(k-1))]);
if(dp[0][t-1]!=temp)
path[0][t-1]=k;
}
cout<<"最短路径长:"<<dp[0][t-1]<<endl;
cout<<"最短路径为: 0";
for(i=t-1,j=0;i>0;)
{
j=path[j][i];
i=i-(1<<(j-1));
cout<<"-->"<<j;
}
cout<<endl;
}