【最小生成树】【并查集】灌水

很明显的最小生成树数…写的时候我竟然一时忆不起来…
还有一不小心思路打对了结果数组开小了交上去掉分…wsl

---

题目描述

Farmer John已经决定把水灌到他的n块农田，农田被数字1到n标记。把一块土地进行灌水有两种方法，从其他农田饮水，或者这块土地建造水库。

建造一个水库需要花费wi,连接两块土地需要花费Pij.

计算Farmer John所需的最少代价。

输入

*第一行：一个数n

*第二行到第n+1行：第i+1行含有一个数wi

*第n+2行到第2n+1行：第n+1+i行有n个被空格分开的数，第j个数代表pij。

输出

*第一行：一个单独的数代表最小代价.

---

Sample-in

4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

Sample-out

9

---

思路

标准最小生成树模板。
关于至少要有一个点建水库：新建一个点0，所有点连接它的权值为建水库的花费。
喔，最小生产树连点这里还用了下并查集。

---

Code

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
int F[10001],n,jl,t,h,ans,f;
struct aaa{
 int xx,yy,zz;
} B[2000001];
int find(int d){  //  并查集
 if(F[d] == d) return d;
 return F[d] = find(F[d]);
}
bool cmp(aaa aa,aaa bb){  //排序：按边的权值从小到大
 return aa.zz<bb.zz;
}
int main(){
 scanf("%d",&n);
 for(int i = 1; i <= n; ++i){
  scanf("%d",&w);
  B[++t] = (aaa){0,i,w};  //存路径，连接0号节点[i] = i; 
 }
 for(int i = 1; i <= n; ++i)
   for(int j = 1; j <= n; ++j){
      scanf("%d",&f);
      if(j>i) B[++t] = (aaa){i,j,f};
   }
 sort(B+1,B+1+t,cmp);  //按权值排序，从小到大
 h=1;
 while(h<=t){  //枚举，连边
  if(find(B[h].xx)!=find(B[h].yy)){  //如果不在一个连通块上
   ans += B[h].zz;  //答案加上
   F[find(B[h].xx)] = F[B[h].yy];  //合并
  }
  ++h;
 }
 printf("%d",ans);  //输出
}