P2916 [USACO08NOV]Cheering up the Cow G 題解
阿新 • • 發佈:2022-04-06
前置知識:最小生成樹演算法(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; }