实际上最终得到的最小生成树,是根节点到达每个点的最短路径。
因为和Dijkstra算法的原理类似,都是局部贪心推最优,且不会出现环。
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
//Max=0x7fffffffffffffff,long long有八个字节16位十六进制,减去一个符号位。
const long long Max=9223372036854775807;
long long cost[1000][1000];
long long mincost[1000000];
bool used[1000000];
int v;
int prim(){
long long res=0;
for(int i=1;i<=v;i++){
mincost[i]=Max;
used[i]=false;
}
mincost[1]=0;
while(true){
int p=-1;
for(int i=1;i<=v;i++){
// 每次都得到used点连接的最短边。
// 局部贪心推全局,万法归贪。
if(!used[i]&&(p==-1||mincost[p]>mincost[i]))p=i;
}
if(p==-1)
break;
used[p]=true;
res+=mincost[p];
for(int j=1;j<=v;j++){
// mincost存储的值是used点能连接上j点的最短边。
mincost[j]=min(mincost[j],cost[p][j]);
}
}
return res;
}
int main(){
// 此处注意初始化不能连通的边。
memset(cost,Max,sizeof(cost));
long long edge;
cin>>v>>edge;
long long l,r;
for(int i=0;i<edge;i++){
cin>>l>>r;
cin>>cost[l][r];
cost[r][l]=cost[l][r];
}
int sum=prim();
cout<<sum;
return 0;
}