链接:点击打开链接
prim模板题,求联通的最小距离。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int INF=0xffffff;
int edge[110][110];
int nn,vis[110];
void prime(int x){
int i,j,k;
int sum_i,mark[110],mi_n;
for(i=1;i<=nn;i++){
mark[i]=edge[x][i];
vis[i]=0;
}
vis[x]=1;
sum_i=0;
for(i=2;i<=nn;i++){
mi_n=INF;
for(j=1;j<=nn;j++){
if(!vis[j]&&mark[j]<mi_n){
mi_n=mark[j];
k=j;
}
}
vis[k]=1;
sum_i+=mi_n;
if(sum_i>=INF)
break;
for(j=1;j<=nn;j++){
if(!vis[j]&&edge[k][j]<mark[j])
mark[j]=edge[k][j];
}
}
printf("%d\n",sum_i);
}
int main(){
int i,j,a;
while(~scanf("%d",&nn)){
for(i=1;i<=nn;i++)
for(j=1;j<=nn;j++)
edge[i][j]=INF;
for(i=1;i<=nn;i++)
for(j=1;j<=nn;j++){
scanf("%d",&a);
edge[i][j]=a;
}
prime(1);
}
return 0;
}