TSP问题——动态规划

问题描述

假设有一个旅行商人要拜访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;
}