Prime(Prim)算法
题目
Problem Description
当前农村公路建设正如火如荼的展开,某乡镇政府决定实现村村通公路,工程师现有各个村落之间的原始道路统计数据表,表中列出了各村之间可以建设公路的若干条道路的成本,你的任务是根据给出的数据表,求使得每个村都有公路连通所需要的最低成本。
Input
连续多组数据输入,每组数据包括村落数目N(N <= 1000)和可供选择的道路数目M(M <= 3000),随后M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个村庄的编号和修建该道路的预算成本,村庄从1~N编号。
Output
输出使每个村庄都有公路连通所需要的最低成本,如果输入数据不能使所有村庄畅通,则输出-1,表示有些村庄之间没有路连通。
Sample Input
5 8
1 2 12
1 3 9
1 4 11
1 5 3
2 3 6
2 4 9
3 4 4
4 5 6
Sample Output
19
代码&分析
对于所有的点,首先加入第一个点,然后找到到这个点a最近的点b,把点b加入进来,这时候生成树为(a,b),然后此时更新lowcost为当前(a,b)到剩下的所有点的权,这样循环,再找到剩下的最短距离的点c,再把c加入进来,此时生成树为(a,b,c),以此类推。
#include<stdio.h>
#include<iostream>
#include<memory.h>
using namespace std;
#define inf 0x3f3f3f3f //地图的初始值
int map[1010][1010]; //(i,j)表示i到j有一条边,权为map[i][j]的值
bool vis[1010]; //标志某个村庄是否已经包括进来
int lowcost[1010]; //对于当前已经加入最小生成树的点的集合,到一个未加入的点的权的数组
int flag;
int sum;
int n;
/*
3362 Prime算法 最小生成树
*/
void prime(){
int temp;
int k;
vis[1]=true;
for(int i=1; i<=n; i++){ //第一次的时候,首先是将lowcost初始化为其他的点到1的最短距离
lowcost[i] = map[i][1];
}
for(int i=2; i<=n; i++){ //共有n个点,要循环n-1次
temp = inf;
for(int j=1; j<=n; j++){ //找到一个未加入最小生成树且权重最小的点
if(!vis[j] && lowcost[j]<temp){
temp = lowcost[j];
k=j;
}
}
if(temp==inf){ //没有找到最小值,说明不连通
flag=1;
break;
}
vis[k]=true; //把权值加上,并且标志为已经加入
sum+=temp;
for(int j=1; j<=n; j++){ //这里更新lowcost,使lowcost总是保存 当前 生成树 到 剩下的所有点 的权
if(!vis[j]&&lowcost[j]>map[j][k])
lowcost[j]=map[j][k];
}
}
}
int main(){
int u, v, m;
int cost;
while(cin>>n>>m){
sum = 0;
flag = 0;
memset(map, inf, sizeof(map)); //初始化为较大的数,一般是定义c++中无穷大的数 0x3f3f3f3f
memset(vis, false, sizeof(vis));
for(int i=0; i<m; i++){
cin>>u>>v>>cost;
map[u][v] = map[v][u] = cost;
}
prime();
if(!flag)
cout<<sum<<endl;
else
cout<<"-1"<<endl;
}
}