bzoj 1601 灌水

题目大意：

决定把水灌到n块农田，农田被数字1到n标记

把一块土地进行灌水有两种方法，从其他农田饮水，或者这块土地建造水库

建造一个水库需要花费wi,连接两块土地需要花费Pij. 计算所需的最少代价

思路：

新开一个节点与每个农田连一条长为wi的边

然后直接最小生成树

 #include<iostream>
 #include<cstdio>
 #include<cmath>
 #include<cstdlib>
 #include<cstring>
 #include<algorithm>
 #include<vector>
 #include<queue>
 #define inf 2139062143
 #define ll long long
 #define MAXN 320
 #define MOD 1000000007
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(!isdigit(ch)) {if(ch=='-') f=-;ch=getchar();}
     while(isdigit(ch)) {x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 int n,cnt,fa[MAXN];
 struct data
 {
     int u,v,val;
     bool operator < (const data &a) const
     {
         return val<a.val;
     }
 }e[MAXN*MAXN];
 int find(int x) {return x==fa[x]?x:fa[x]=find(fa[x]);}
 bool ok(int u,int v)
 {
     int fu=find(u),fv=find(v);
     if(fu==fv) return ;
     else {fa[fu]=fv;return ;}
 }
 void add(int u,int v,int w)  {e[++cnt].u=u,e[cnt].v=v,e[cnt].val=w;}
 void kruskal()
 {
     sort(e+,e+cnt+);
     int res=,tot=;
     for(int i=;i<=cnt;i++)
     {
         if(!ok(e[i].u,e[i].v)) res+=e[i].val,tot++;
         if(tot==n) break;
     }
     printf("%d",res);
 }
 int main()
 {
     n=read();int a,b,c;
     for(int i=;i<=n;i++) {a=read(),fa[i]=i;add(i,n+,a);}
     for(int i=;i<=n;i++)
         for(int j=;j<=n;j++)
         {
             a=read();
             if(i!=j) add(i,j,a);
         }
     kruskal();
 }