HDU 1301-Jungle Roads prime算法的典型应用

最新推荐文章于 2020-04-22 12:14:14 发布

昧者-卑贱的湘伦

最新推荐文章于 2020-04-22 12:14:14 发布

阅读量541

点赞数

分类专栏：代码攻略文章标签： prime算法最小生成树模板

本文链接：https://blog.csdn.net/xianglunxi/article/details/9016685

版权

代码攻略专栏收录该内容

57 篇文章 0 订阅

订阅专栏

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1301

参考代码：

#include<iostream>
#include<stdio.h>
#include<memory.h>
using namespace std;
#define INF 0x7ffffff
#define MAXN 105
int map[MAXN][MAXN],lowcost[MAXN];
bool visit[MAXN];
int n,sum;

int prim()
{
	int temp,k;
	sum = 0;
	memset(visit, false, sizeof(visit)); //初始化visit
	for(int i = 1;i <= n;i++) //初始化lowcost[i]
		lowcost[i] = map[1][i];
     visit[1]=true;//第一个顶点标记已访问
	for(int i = 2;i <= n;i++)//找生成树集合点集相连最小权值的边
	{
		temp = INF;
		for(int j = 1; j <= n; j++)
			if(!visit[j] && temp > lowcost[j])
				temp = lowcost[k = j];
		if(temp == INF) break;
		visit[k] = true; //加入最小生成树集合
		sum += temp;//记录权值之和
		for(int j = 1; j <= n; j++) //更新lowcost数组
			if(!visit[j] && lowcost[j] > map[k][j])
				lowcost[j] = map[k][j];
	}
return sum;
}

int main()
{
    char c1,c2;
    int road,cost,i,j,a,b;
    while(scanf("%d", &n)&&n)
	{
		//memset(map, INF, sizeof(map));//一定不能这样写，否则答案是错误的。不知道是为什么
		for(i=1;i<=n;i++)
          for(j=1;j<=n;j++)
		 {
		    map[i][j]=INF;
		 }
       for(i=1;i<n;i++)
        {
          getchar();
          scanf("%c%d",&c1,&road);
          for(j=1;j<=road;j++)
          {
            getchar();
            scanf("%c%d",&c2,&cost);
            a=c1-'A'+1;//从点1开始
            b=c2-'A'+1;//从点1开始
            map[a][b] = map[b][a] =cost;
          }
        }
	printf("%d\n",prim()); //最小生成树权值之和
	}
	return 0;
}