加油。
To be better.
题目链接:
https://cn.vjudge.net/problem/HDU-1233
题目大意就是给你n个城市,给你路径,然你用n-1条路将n个城市连接起来。保证所有公路的长度最短。
这里采用 Prime算法
首先解释一下什么是Prime算法
在带权连通图中V是包含所有顶点的集合,U是已经在最小生成树中的节点;
1.初始时,从图中任意某一顶点v开始,此时集合U={v}(以v到其他顶点的所有边为侯选边);
2. 在所有u∈U,w∈V-U的边(u,w)∈E中找到一条权值最小的边,将(u,w)这条边加入到已找到边的集合,并且将点w加入到集合U中,
3.重复上一操作,当U=V时,就找到了这颗最小生成树。
核心步骤:在所有u∈U,w∈V-U的边(u,w)∈E中找到一条权值最小的边。
下面如果遇到不懂,就再回到这里仔细品读Prime算法概述
按照题目样例讲解:
题目样例:
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
首先可以在演草纸上画一个这样的图:
然后可以用图的邻接矩阵存到一个数组map中
map[i][j]数组
1 | 2 | 3 | 4 | |
---|---|---|---|---|
1 | 0 | 1 | 4 | 1 |
2 | 1 | 0 | 3 | 2 |
3 | 4 | 3 | 0 | 5 |
4 | 1 | 2 | 5 | 0 |
详解代码:
#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define MAX 105
#define INF 9999999
int map[MAX][MAX];//记录两点间权值的数组
bool vis[MAX];//记录该点是否访问过
int dis[MAX];//记录点到目标集合的最小权值
int N;
int sum;
void Prim()
{//普里姆算法
//默认1为起点
int i,j,v,Min;
sum=0;
for(i=2; i<=N; i++)
{//将起点到各个点的距离保存
dis[i]=map[1][i];
vis[i]=0;//初始化所有点都没访问
}
dis[1]=0;//起点到起点本身,距离为0
vis[1]=1;//从1开始访问
for(i=2; i<=N; i++)
{
Min=INF;
for(j=2; j<=N; j++)
{
if(!vis[j]&&dis[j]<Min)
{
Min=dis[j];
v=j;
}
}
vis[v]=1;//将找到的点加入最小生成树中
sum+=dis[v];
for(j=2; j<=N; j++)
{//更新不在最小生成树中的点的边权值
if(!vis[j]&&dis[j]>map[v][j])
dis[j]=map[v][j];
}
}
printf("%d\n",sum);
return;
}
int main()
{
int i,j,a,b,c,S;
while(scanf("%d",&N)!=EOF,N)
{
for(i=1; i<=N; i++)
{
map[i][i]=0;
}
S=N*(N-1)/2;
for(i=1; i<=S; i++)
{
scanf("%d%d%d",&a,&b,&c);
map[a][b]=map[b][a]=c;
}
Prim();
}
return 0;
}