蓝桥杯 算法训练 ALGO-6 安慰奶牛

#include<iostream>
#include<algorithm>
#include<math.h>
using namespace std;

int pre[11000];
int point[11000];
struct Edge
{
int x,y,w;
}edge[110000];
int n,m;

int cmp(Edge a,Edge b)
{
return a.w<b.w;
}

int find(int x)
{
if(pre[x]==x)
return x;
int t=find(pre[x]);
pre[x]=t;
return t;
}

int main()
{
int i;
while( scanf("%d%d",&n,&m)!=EOF )
{
int mmin=10000;
for(i=1;i<=n;i++)
{
pre[i]=i;
scanf("%d",&point[i]);
mmin=min(mmin,point[i]);
}

for(i=1;i<=m;i++)
{
scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].w);
edge[i].w=edge[i].w*2+point[edge[i].x]+point[edge[i].y];
}

sort(edge+1,edge+m+1,cmp);

int sum=0;
for(i=1;i<=m;i++)
{
int fx=find(edge[i].x);
int fy=find(edge[i].y);
if(fx!=fy)
{
sum=sum+edge[i].w;
pre[fx]=fy;
}
}

printf("%d\n",sum+mmin);
}

return 0;
}