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=1000000000;
int tot,In[1000005],vis[1000005],pre[1000005],id[1000005];
int X,Y,Z,n;
struct Point{
    int x,y,z;
}p[1000005];
struct edge{
    int u,v,w;
}e[1300005];
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=0;
    while (1){
        for (int i=0;i<n;i++) In[i]=inf;
        for (int i=1;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=0;i<n;i++){
            if (i==rt) continue;
            if (In[i]==inf) return -1;
        }
        int cnt=0;
        for (int i=0;i<n;i++) vis[i]=id[i]=-1;
        In[rt]=0;
        for (int i=0;i<n;i++){
            res+=In[i];
            int v=i;
            while (vis[v]!=i&&id[v]==-1&&v!=rt){
                vis[v]=i;
                v=pre[v];
            }
            if (v!=rt&&id[v]!=-1){
                for (int j=pre[v];j!=v;j=pre[j]){
                    id[j]=cnt;
                }
                id[v]=cnt++;
            }
        }
        if (cnt==0) return res;
        for (int i=0;i<n;i++) if (id[i]==-1) id[i]=cnt++;
        int all=0;
        for (int i=1;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 0;
        for (int i=1;i<=n;i++) scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].z);
        for (int i=1;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(0,i,p[i].z*X);        
        }
        n++;
        int Ans=directedMST(0);
        printf("%d\n",Ans);
    }
}