DFS实现全排列，同时剪枝求解TSP问题

#include<bits/stdc++.h>
using namespace std;
#define MAX 1000
int visit[MAX]={};
double minlen=10000000;
int n;
vector<vector<int> > permute; 
vector<int> vec;
vector<pair<int,int> > vt;
double computelen(pair<int,int> a,pair<int,int> b){
	return sqrt((a.first-b.first)*(a.first-b.first)+(a.second-b.second)*(a.second-b.second));
}
void dfs(int num,int index,int dis){
	if(num==n-1){
		permute.push_back(vec);
		if(minlen>(dis+computelen(vt[0],vt[index]))){
			minlen=dis+computelen(vt[0],vt[index]);
			return;
		}else{
			return;
		}
	}
	if(dis>minlen) return;//剪枝 
	for(int i=0;i<n;i++){
		if(visit[i]==0){
			visit[i]=1;
			vec.push_back(i);
			dfs(num+1,i,dis+computelen(vt[i],vt[index]));
			vec.pop_back();
			visit[i]=0;
			
		}
	}
}
int main(){
	freopen("D:\\Users\\sgcai\\Downloads\\devcpp_downcc\\in.txt","r",stdin);
	cin>>n;
	vt.resize(n);
	for(int i=0;i<n;i++){
		cin>>vt[i].first>>vt[i].second;
	}
	visit[0]=1;
	dfs(0,0,0);
	cout<<minlen<<endl;
	for(int i=0;i<permute.size();i++){
		for(int j=0;j<permute[i].size();j++){
			cout<<permute[i][j]<<' ';
		}
		cout<<endl;
	}
	return 0;
}

题目是这样的：

平原上，一群蜜蜂采蜜，要连续采集5片花丛后归巢，已知5片花丛相对蜂巢的坐标，请你规划一下到访花丛的顺序，以使飞行总距离最短。

输入：以蜂巢为坐标原点的五片花丛A、B、C、D、E的坐标，坐标值为整数

输出：从出发到返回蜂巢的最短路径的长度，长度取整。

样例：

输入：

5
200 0
200 10
200 50
200 30
200 25

下面是剪枝之后的结果：

下面是未剪枝的结果：

分析：

这里只是写思路，并未考虑到一些细节，比如取整与否未考虑；

这道题可以用状态dp来做，也可以直接求出来全排列来做，只是个人复习到DFS算法，故这里用了DFS+剪枝；

函数的各项参数的定义如下：

DFS（num，index，dis）

num表示当前已访问的点的数量；

index表示当前正在访问的点的下标；

dis表示已访问的点集中各点间的距离，最后一次计算为当前正在访问的点和上一步访问的点之间的距离；

到边界之后由于要返回到起点，故最后要记得加上起点和最后访问的那个点之间的距离；

求全排列的过程可以顺便加进去，设两个数组permute和vec，其中vec存储每次完整访问过程的一个排列，permute存储所有的排列组合，简单起见这里没把起点0考虑进去；