BZOJ 1601 [Usaco2008 Oct]灌水：最小生成树

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1601

题意：

　　Farmer John已经决定把水灌到他的n(1<=n<=300)块农田，农田被数字1到n标记。

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

　　建造一个水库需要花费wi(1<=wi<=100000)，连接两块土地需要花费P[i][j](1 <= p[i][j] <= 100000, p[i][j]=p[j][i], p[i][i]=0)。

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

题解：

　　农田标号1到n，建立超级源点为0号点。

　　从超级源点向每个农田连一条长度为w[i]的边。然后n个农田之间再互相连接，边长为p[i][j]。

　　然后求最小生成树就好。

AC Code:

 #include <iostream>
 #include <stdio.h>
 #include <string.h>
 #include <algorithm>
 #include <vector>
 #define MAX_N 305

 using namespace std;

 struct Edge
 {
     int sour;
     int dest;
     int len;
     Edge(int _sour,int _dest,int _len)
     {
         sour=_sour;
         dest=_dest;
         len=_len;
     }
     Edge(){}
     friend bool operator < (const Edge &a,const Edge &b)
     {
         return a.len<b.len;
     }
 };

 int n;
 int ans;
 int par[MAX_N];
 vector<Edge> edge;

 void init_union_find()
 {
     for(int i=;i<=n;i++)
     {
         par[i]=i;
     }
 }

 int find(int x)
 {
     return par[x]==x?x:par[x]=find(par[x]);
 }

 void unite(int x,int y)
 {
     int px=find(x);
     int py=find(y);
     if(px==py) return;
     par[px]=py;
 }

 bool same(int x,int y)
 {
     return find(x)==find(y);
 }

 int kruskal()
 {
     init_union_find();
     sort(edge.begin(),edge.end());
     int cnt=;
     int res=;
     for(int i=;i<edge.size();i++)
     {
         Edge temp=edge[i];
         if(!same(temp.sour,temp.dest))
         {
             cnt++;
             res+=temp.len;
             unite(temp.sour,temp.dest);
         }
     }
     return cnt==n?res:-;
 }

 void read()
 {
     cin>>n;
     int a;
     for(int i=;i<=n;i++)
     {
         cin>>a;
         edge.push_back(Edge(,i,a));
     }
     for(int i=;i<=n;i++)
     {
         for(int j=;j<=n;j++)
         {
             cin>>a;
             if(i<j) edge.push_back(Edge(i,j,a));
         }
     }
 }

 void solve()
 {
     ans=kruskal();
 }

 void print()
 {
     cout<<ans<<endl;
 }

 int main()
 {
     read();
     solve();
     print();
 }