HDU 1223 还是畅通工程（最小生成树prim模板）

一个很简单的prim模板，但虽然是模板，但也是最基础的，也要脱离模板熟练打出来
后期会更新kruskal写法#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn = 1000;
const int INF = 0x3f3f3f3f;//int型最大值
int Map[maxn][maxn],vis[maxn];
int n,dis[maxn],ans;

void init(){
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
if(i==j)Map[i][j]=0;
else Map[i][j]=INF;
}
}
ans=0;
memset(vis,0,sizeof(vis));
}

void input(){
int u,v,w;
for(int i=0;i<n*(n-1)/2;i++){
scanf("%d%d%d",&u,&v,&w);
if(Map[u][v]>w)
Map[u][v]=Map[v][u]=w;
}
}

int prim(){
for(int i=1;i<=n;i++){
dis[i]=Map[1][i];
}
vis[1]=1;
for(int i=1;i<n;i++){
int MIN=INF,x=-1;
for(int j=1;j<=n;j++){
if(!vis[j]&&dis[j]<MIN){//每一次都取该结点和其他相邻的结点的最小值
MIN=dis[j];
x=j;
}
}
if(x==-1)return -1;
ans+=MIN;
vis[x]=1;//标记已经过的节点
for(int j=1;j<=n;j++){
if(!vis[j]&&dis[j]>Map[x][j]){//更新dis数组，
dis[j]=Map[x][j];
}
}
}
}

int main(){
while(scanf("%d",&n)&&n){
init();//初始化
input();//输入
prim();//关键
printf("%d\n",ans);
}
return 0;
}