『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 */