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

#include
using namespace std;
#define MAX 1000
int visit[MAX]={};
double minlen=10000000;
int n;
vector > permute; 
vector vec;
vector > vt;
double computelen(pair a,pair 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;
	vt.resize(n);
	for(int i=0;i>vt[i].first>>vt[i].second;
	}
	visit[0]=1;
	dfs(0,0,0);
	cout<

题目是这样的:

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

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

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

样例:

输入:

5
200 0
200 10
200 50
200 30
200 25

下面是剪枝之后的结果:

DFS实现全排列,同时剪枝求解TSP问题_第1张图片

下面是未剪枝的结果:

DFS实现全排列,同时剪枝求解TSP问题_第2张图片

分析:

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

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

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

DFS(num,index,dis)

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

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

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

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

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

你可能感兴趣的:(C++/PAT/洛谷OJ)