201412-4-最优灌溉

#include<cstdio>
#include<vector>
#include<cstring>
#define INF 0xfffffff
#define maxn 1005
using namespace std;
int dis[maxn],edge[maxn][maxn];
bool vis[maxn];
int n,m,u,v,w,mi;
void Prim()//点归并
{
int i,j,k,ans=0;
memset(vis,0,sizeof(vis));
for(i=1;i<=n;i++)
dis[i]=edge[1][i];
vis[1]=1;
for(i=1;i<n;i++)
{
mi=INF;
for(j=1;j<=n;j++)
{
if(!vis[j]&&dis[j]<mi)
{
mi=dis[j];
k=j;
}
}
vis[k]=1;
ans+=mi;//ans为距离之和
for(j=1;j<=n;j++)
if(!vis[j]&&dis[j]>edge[k][j])
dis[j]=edge[k][j];
}
printf("%d\n",ans);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
edge[i][j]=INF;//非邻接顶点距离用INF标记
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&u,&v,&w);
edge[u][v]=w;
edge[v][u]=w;
}
Prim();
return 0;
}