1. 程式人生 > 其它 >P2916 [USACO08NOV]Cheering up the Cow G 題解

P2916 [USACO08NOV]Cheering up the Cow G 題解

前置知識:最小生成樹演算法(Kruskal/Prim)

例題

演算法分析:

這一道題中給出一個無向圖,求從任意一點開始經過每一點的最短路徑。

既然要經過每一個點,還要求最短路徑,演算法就是最小生成樹了。

我用的是 Kruskal 演算法。

有一點需要注意:每條路的長度需要如何計算?

約翰需要經過每一條道路兩次,並且每走一次,就需要安慰這條路後的奶牛,所以路徑長度=道路長度*2+安慰兩頭奶牛時間

程式碼如下:(因為幾乎是 Kruskal 模板,所以沒有註釋)

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
struct node
{
 LL x,y,dis;
}a[100000+10];
LL n,p,c[10000+10],fa[10000+10],t,ans;
bool cmp(node fir,node sec)
{
 return fir.dis<sec.dis;
}
LL gf(LL x)
{
 if(fa[x]==x) return x;
 return fa[x]=gf(fa[x]);
}
int main()
{
 scanf("%lld%lld",&n,&p);
 for(int i=1;i<=n;i++) scanf("%lld",&c[i]);
 for(int i=1;i<=p;i++)
 {
  scanf("%lld%lld%lld",&a[i].x,&a[i].y,&a[i].dis);
  a[i].dis=a[i].dis*2+c[a[i].x]+c[a[i].y];
 }
 for(int i=1;i<=n;i++) fa[i]=i;
 sort(a+1,a+p+1,cmp);
 for(int i=1;t<n-1;i++)
 {
  int fx=gf(a[i].x);
  int fy=gf(a[i].y);
  if(fx!=fy)
  {
   fa[fx]=fy;
   ans+=a[i].dis;
   t++;
  }
 }
 sort(c+1,c+n+1);
 printf("%lld\n",ans+c[1]);
 return 0;
}