那道題搞一搞會發現在選好的n-1 條邊上,每條邊走兩次,還需要花掉兩個端點需要的時間,才能回到原點。那麼我們不妨把每個邊權設成路長*2+兩個端點的時間,這樣建最小生成樹就可以了。還有,回到出發點還要加上出發點的值。
#include<stdio.h>
#include<algorithm>
using namespace std;
struct node
{
int from;
int to;
int len;
}edge[2000000];
int cmp(node a,node b)
{
return a.len<b.len?1:0;
}
int n,m;
int val[2000000];
int f[2000000];
int ans;
void in()
{
for(int i=1;i<=n;i++)
{
f[i]=i;
}
}
int find(int x)
{
if(f[x]==x)
{
return x;
}else
{
f[x]=find(f[x]);
return f[x];
}
}
void kru()
{
int cnt=0;
for(int i=1;i<=m;i++)
{
int fx=find(edge[i].from);
int fy=find(edge[i].to);
if(fx!=fy)
{
cnt++;
f[fy]=fx;
ans+=edge[i].len;
if(cnt==n-1)
{
return ;
}
}
}
}
int main()
{
scanf("%d %d",&n,&m);
int min=9999999;
for(int i=1;i<=n;i++)
{
scanf("%d",&val[i]);
if(val[i]<min)
{
min=val[i];
}
}
for(int i=1;i<=m;i++)
{
int v;
scanf("%d %d %d",&edge[i].from,&edge[i].to,&v);
edge[i].len=val[edge[i].from]+val[edge[i].to]+v*2;
}
sort(edge+1,edge+1+m,cmp);
in();
kru();
printf("%d",ans+min);
}