HDU - 2489详细题解

Prim算法与DFS搜索综合运用实例：HDU - 2489（详解）

       本题即是要求一个比例的最小值，只不过这个比例是相当于对于一个树的子树，其边的权重之和与节点权重之和的比值。在主函数中，我们首先输入节点的权重，然后运用二维数组保存边，输入的是从第几个节点开始的第几条边的权重（邻接矩阵）（按照数字从小到大的顺序规律性地存储）。

       紧接着，完成了录入操作之后，首先第一步，就是要枚举出从n个点中选出m个点的所有不同情况，然后对这m个点求其最小生成树，此时即为答案（因为选出了m个点之后，就相当于其节点本身的权重之和就已经固定了，此时只需要边的权重最小即可）。

       对于这个枚举操作的实现，这里使用的是DFS，即最终我们要得到的则是DFS(n,m)。由于DFS运用的思想实则为递归，因此先将vis数组初始化为0后，先给最小比例赋一个很大的值，然后进行枚举。我们首先传递1、0进去，此时就需要进行if判断，假设形参分别为u、num，首先判断是否num>m，若是则立即return。紧接着，1：如果u==n+1，那么当num!=m时，此时也直接返回。然后就是找最小比例minRatio了，令double t=prim()（其中那个函数是核心算法），如果t<minRatio，那么就令minRatio = t，然后将vis数组整体复制给ans数组，然后return（这里的ans数组是为之后的输出而用的，因为我们只会输出被访问过的节点的编号）；2、如果不存在u==n+1，那么就使得vis[u] =1，然后进行递归DFS(u+1,num+1)，并且由于在之后还有其它情况的递归过程，因此在此轮递归完成后需要复原（即vis[u] =0），由于此时只能遍历u比num大1的情况，因此当实参n-1>=m时，我们就需要再进行dfs(u+1,num)的递归操作，这样就能遍历全部的情况了。得到了最小比例后，就需要对编号进行输出，先设定一个初始值为0的标记变量flag，对于每一个访问过的节点，若此时flag为1，则先输出一个空格，再输出节点的编号；若flag为0，则说明此时为第一个要输出的值，因此直接输出节点编号，然后令flag = 1即可。最后输出一个换行符，整个程序结束。

       不过，本题有一个核心算法，就是求那个最小比例的值的时候所用到的Prim算法，此算法也是用来解决求最小生成树的问题的，其大致的步骤，首先是输入一个加权连通图，其中包括顶点集合V和边集合E；然后初始化：Vnew = {x}（其中x为集合V中的任一节点（起始点）），Enew为空；然后重复进行接下来的两个操作直至Vnew = V：1、在集合E中选取权值最小的边<u, v>，其中u为集合Vnew中的元素，而v不在Vnew集合当中，并且v∈V（如果存在有多条满足前述条件即具有相同权值的边，则可任意选取其中之一），2、将v加入集合Vnew中，将<u, v>边加入集合Enew中；最后就是输出，使用集合Vnew和Enew来描述所得到的最小生成树。在本题当中，hash数组即为点集，首先我们从头开始遍历点，若碰到第一个已经访问过的点时，则将其编号用变量u临时记录下来，然后退出遍历，此时将hash[u] =1（即我们所选的初始电即为第一个被访问过的点），然后重新开始遍历，碰到访问过的点i时，就让minCost[i] = G[u][i]（其中minCost[i]表示以i为终点的边的最小权值），再让pre[i] = u（即这条以i为终点的边的起点点编号），然后weightSum += weight[i]（将点总权重更新），因此，这第一次循环就是计算题目所要计算的分母的。接下来进行第二次循环，每一次循环前都先让u = -1，然后即使有一个for语句进行遍历，当j点已经被访问过（此处所指的访问过实则指的是选中的那几个点）而且j还未进入点集时（即!hash[j]），此时再判断如果u==-1 或者是当 minCost[u]>minCost[j]时，就让u = j（此即为权值最小的边，因为编号都是按照从小到大的顺序来的，因此此时找的即为正确的距离点最近而且权值最小的边），然后G[pre[u]][u]那条边就是我们要选的边，将其加到分子中，此时再让hash[u] =1。随后，就再进行一次遍历（这里是一个大的遍历里面包含两个小的遍历），此时当满足vis[j]&&!hash[j]（此时除去了刚刚在上一个遍历中找到的u点）时，若minCost[j] > G[u][j]，则将值更新：minCost[j] = G[u][j]，pre[j] = u。（这说明，上一个小的遍历是找在DFS遍历中选出来的所含有的最小的点的最小的边，然后接下来一个小的遍历是对剩下的选中的点（对于刚刚连接了的点u来说即为下一个将要与u连接的点）的以其为终点的边的权重最小值，即其在minCost数组里所代表的值。然后接下来再进行一次大的遍历时，就会找到那个点。由于那个点的minCost数组里的值刚才被更新了，因此就是最小值。）（先确定一个起始点，通过一个遍历找经过它的最小的一条边，然后再继续一个个地向下延伸。）此即为一个大的遍历当中的一整步，接下来再进行下一次的大的遍历。最后进行完成后，返回的edgeSum/weightSum值即为所要求的minRatio。

       这里附上一个AC代码：

#include <stdio.h>
#include <string.h>
int m,n,vis[20],ans[20],pre[20],hash[20];
double G[20][20],weight[20],minCost[20],minRatio;
double prim()
{
	memset(hash,0,sizeof(hash));
	int u;
	for(int i=1;i<=n;++i)
		if(vis[i])
		{
			u=i;
			break;
		}
	hash[u] = 1;
	double weightSum=0,edgeSum=0;
	for(int i=1; i<=n; ++i)if(vis[i])
	{
		minCost[i] = G[u][i];
		pre[i] = u;
		weightSum += weight[i];
	}
	for(int i=1; i<m; ++i)
	{
		u=-1;
		for(int j=1; j<=n; ++j)if(vis[j]&&!hash[j])
			if(u==-1 || minCost[u]>minCost[j])
				u=j;
		edgeSum += G[pre[u]][u];
		hash[u] = 1;
		for(int j=1; j<=n; ++j)
			if(vis[j]&&!hash[j])
				if(minCost[j] > G[u][j])
				{
					minCost[j] = G[u][j];
					pre[j] = u;
				}
	}
	return edgeSum/weightSum;
}
void dfs(int u,int num)
{
	if(num>m)
		return;
	if(u==n+1)
	{
		if(num!=m)
			return;
		double t=prim();
		if(t<minRatio)
		{
			minRatio = t;
			memcpy(ans,vis,sizeof(vis));
		}
		return;
	}
	vis[u] = 1;
	dfs(u+1,num+1);
	vis[u] = 0;
	if(n-1>=m)
		dfs(u+1,num);
}
int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		if(m==0&&n==0)
			break;
		for(int i=1;i<=n;++i)
			scanf("%lf",&weight[i]);
		for(int i=1;i<=n;++i)
			for(int j=1;j<=n;++j)
				scanf("%lf",&G[i][j]);
		memset(vis,0,sizeof(vis));
		minRatio = 100000000;
		dfs(1,0);
		int flag=0;
		for(int i=1; i<=n; ++i)
			if(ans[i])
			{
				if(flag)
					printf(" %d", i);
				else
				{
					printf("%d",i);
					flag=1;
				}
			}
		printf("\n");
	}
	return 0;
}