1. 程式人生 > >『USACO08OCT]』Watering Hole

『USACO08OCT]』Watering Hole

題面

code

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 int n,cnt,ans,fa[305];
 4 struct edge{int x,y,v;}e[100005];
 5 inline int read() {
 6     int x=0,f=1; char c=getchar();
 7     while(c<'0'||c>'9') {if(c=='-')f=-1; c=getchar();}
 8     while(c>='0'&&c<='9') x=(x<<3
)+(x<<1)+c-'0',c=getchar(); 9 return x*f; 10 } 11 int getfa(int x) {return x==fa[x]?x:fa[x]=getfa(fa[x]);} 12 int add(int x,int y,int z) { 13 e[++cnt].x=x; e[cnt].y=y; e[cnt].v=z; 14 } 15 int cmp(edge x1,edge x2) {return x1.v<x2.v;} 16 17 int main() { 18 n=read(); 19 for(int
i=1;i<=n;i++) fa[i]=i; 20 for(int i=1;i<=n;i++) {int vv=read(); add(0,i,vv);} 21 for(int i=1;i<=n;i++) 22 for(int j=1;j<=n;j++) { 23 int vv=read(); 24 if(i!=j) add(i,j,vv); 25 } 26 sort(e+1,e+cnt+1,cmp); 27 for(int i=1;i<=cnt;i++) { 28 int
x=getfa(e[i].x),y=getfa(e[i].y); 29 if(x!=y) {cnt++; ans+=e[i].v; fa[x]=y;} 30 if(cnt==n) break; 31 } 32 printf("%d",ans); 33 } 34 /* 35 #最小生成樹# 36 將打井看做與一個新的節點(0)連線 ,邊權為 w[i] 37 相當於圖中有n+1個節點 ,求它們的最小生成樹。 38 */