HDU 4009 Transfer water

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4009

题意：给出一个村庄(x,y,z)。每个村庄可以挖井或者修建水渠从其他村庄得到水。挖井有一个代价，修水渠有一个代价。另外A村庄只能向其指定的一些村庄供水。使得所有村庄有水求最小代价。

思路，建立源节点0，向每个点连边视为挖井，如果从其他村庄得到水就相当于从那个点引一条有向边过来。

 #include<algorithm>
 #include<cstdio>
 #include<cmath>
 #include<cstring>
 #include<iostream>
 const int inf=;
 int tot,In[],vis[],pre[],id[];
 int X,Y,Z,n;
 struct Point{
     int x,y,z;
 }p[];
 struct edge{
     int u,v,w;
 }e[];
 void insert(int x,int y,int z){
     tot++;
     e[tot].u=x;
     e[tot].v=y;
     e[tot].w=z;
 }
 int dis(Point p1,Point p2){
     int ans=std::abs(p1.x-p2.x)+std::abs(p1.y-p2.y)+std::abs(p1.z-p2.z);
     ans*=Y;
     if (p1.z<p2.z) ans+=Z;
     return ans;
 }
 int directedMST(int rt){
     int res=;
     while (){
         for (int i=;i<n;i++) In[i]=inf;
         for (int i=;i<=tot;i++){
             int u=e[i].u,v=e[i].v;
             if (u!=v&&In[v]>e[i].w) In[v]=e[i].w,pre[v]=u;
         }
         for (int i=;i<n;i++){
             if (i==rt) continue;
             if (In[i]==inf) return -;
         }
         int cnt=;
         for (int i=;i<n;i++) vis[i]=id[i]=-;
         In[rt]=;
         for (int i=;i<n;i++){
             res+=In[i];
             int v=i;
             while (vis[v]!=i&&id[v]==-&&v!=rt){
                 vis[v]=i;
                 v=pre[v];
             }
             if (v!=rt&&id[v]!=-){
                 for (int j=pre[v];j!=v;j=pre[j]){
                     id[j]=cnt;
                 }
                 id[v]=cnt++;
             }
         }
         if (cnt==) return res;
         for (int i=;i<n;i++) if (id[i]==-) id[i]=cnt++;
         int all=;
         for (int i=;i<=tot;i++){
             int v=e[i].v,u=e[i].u;
             e[i].u=id[u];
             e[i].v=id[v];
             if (e[i].u!=e[i].v)
              e[i].w-=In[v],e[++all]=e[i];
         }
         tot=all;
         n=cnt;rt=id[rt];
     }
     return res;
 }
 int main(){
     while (scanf("%d%d%d%d",&n,&X,&Y,&Z)!=EOF&&n){
         if (!n&&!X&&!Y&&!Z) return ;
         for (int i=;i<=n;i++) scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].z);
         for (int i=;i<=n;i++){
             int k,x;
             scanf("%d",&k);
             while (k--){
                 scanf("%d",&x);
                 if (x==i) continue;
                 insert(i,x,dis(p[i],p[x]));
             }
             insert(,i,p[i].z*X);
         }
         n++;
         int Ans=directedMST();
         printf("%d\n",Ans);
     }
 }