旅行商问题

问题描述

假设有一个货郎担要拜访n个城市，他必须选择所要走的路程，路程的限制时每个城市只能拜访一次，而且最后要走到原来出发的城市，要求路径长度。

解法一：bfs

#include <bits/stdc++.h>
using namespace std;
const int N=100;
int n,m,g[N][N],d,num,r[N];
bool st[N];
vector<int> p;
vector<vector<int>> q;
bool check()
{
    for(int i=1;i<n;i++)
        if(!st[i]) return false;
        
    return true;
}
void bfs(int x)
{
//判断是否已经完全走完所有的点
    if(check())
    {
        if(g[x][0])
        {
            q[num].push_back(0);
//符合条件的就将路径保存
            for(int i=0;i<p.size();i++)    q[num].push_back(p[i]);
            
            q[num].push_back(0);
            d+=g[x][0];
//保存路径长度
            r[num]=d;
            d-=g[x][0];
            num++;
        }
        return ;
    }
    for(int i=1;i<n;i++)
    {
        if(g[x][i]&&!st[i])
        {
            p.push_back(i);
            d+=g[x][i];
            st[i]=1;
            
            bfs(i);//进行递归处理
//对变量进行复原         
            st[i]=0;
            d-=g[x][i];
            p.pop_back();
        }
    }
}
int main()
{
    q.resize(110);
//输入点、边和权值
    cin>>n>>m;
    for(int i=0;i<m;i++)
    {
        int x,y,z;
        cin>>x>>y>>z;
        g[x][y]=z;
    }
	 
    bfs(0);
//对最短路径进行查询
    int t=0;
    for(int i=0;i<num;i++)
        if(r[i]<r[t]) t=i;
        
    for(int i=0;i<=n;i++)    cout<<q[t][i]<<" ";
    cout<<r[t];

    
    return 0;
}

解法二：dp
 

#include <bits/stdc++.h> 
using namespace std;
const int N = 21, M = 1 << N;
typedef long long ll;
ll dp[M][N];
ll w[N][N];
int n,m;
vector<int> path[N][N];
int main()
{
    cin>>n>>m;
    memset(w,0x3f,sizeof w);
    
    for(int i=0;i<m;i++)
    {
    	int x,y,z;
    	cin>>x>>y>>z;
    	w[x][y]=z;
	}
            
    memset(dp, 0x3f, sizeof dp);
    dp[1][0] = 0;
    
    for (int i = 0; i < 1 << n; i ++)
    {
    	for (int j = 0; j < n; j ++)
    	{
    		if (i >> j & 1)
    		{
    			for (int k = 0; k < n; k ++ )
    			{
    				if (i >> k & 1)
                    {
                    	if(dp[i][j]>dp[i - (1 << j)][k] + w[k][j])
                    	{
                    		dp[i][j] =dp[i - (1 << j)][k] + w[k][j];
                    		path[i][j] = path[i - (1 << j)][k]; 
                        	path[i][j].push_back(j); 
						}
					}
				}
                    
			}
                
		}
            
	}
    //得到从0到每个点之间的最短路
    
	int t=1,num=dp[(1<<n)-1][1]+w[1][0]; 
	for(int i=2;i<n;i++)
	{
		if(dp[(1<<n)-1][i]+w[i][0]<num) 
		{
			num=dp[(1<<n)-1][i]+w[i][0];
			t=i;
		}
	}				
    cout<<"0 ";                    
    for(int i=0;i<path[(1<<n)-1][t].size();i++) cout<<path[(1<<n)-1][t][i]<<" ";    
	cout<<"0 "<<num;          
	    return 0;
}

解法三：剪枝后的bfs

#include <bits/stdc++.h>
using namespace std;
const int N=100;
int n,m,g[N][N],d;
bool st[N];
vector<int> p,q;
set<int> s; 
bool check()
{
    for(int i=1;i<n;i++)
        if(!st[i]) return false;
        
    return true;
}
void bfs(int x)
{
    if(check())
    {
        if(g[x][0])
        {
            
            d+=g[x][0];
            
            if(s.size()&&d>*s.begin()) 
			{
            	d-=g[x][0];
            	return ;
			}
    
	        s.insert(d);
	        
	        q.clear();
	        q.push_back(0);
	        for(int i=0;i<p.size();i++)    q.push_back(p[i]);
	        q.push_back(0);
			
			d-=g[x][0];
        }
        return ;
    }
    
    for(int i=1;i<n;i++)
    {
        if(g[x][i]&&!st[i])
        {
            p.push_back(i);
            d+=g[x][i];
            st[i]=1;
            
            if(s.size()&&d>*s.begin()) return ;
            
            bfs(i);
            
            st[i]=0;
            d-=g[x][i];
            p.pop_back();
        }
    }
}
int main()
{
    q.resize(110);
    cin>>n>>m;
    for(int i=0;i<m;i++)
    {
        int x,y,z;
        cin>>x>>y>>z;
        g[x][y]=z;
    }
	 
    bfs(0);
   
   for(int i=0;i<q.size();i++) cout<<q[i]<<" ";
   cout<<*s.begin();

    
    return 0;
}