poj 1258 Agri-Net

链接：点击打开链接

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;
}