hdu 4009 Transfer water

http://acm.hdu.edu.cn/showproblem.php?pid=4009

最小树形图。题意是有n个地方需要供水，每个地方都可以选择是自己挖井，还是从别的地方引水，根据方法不同和每个地方的坐标不同，花费也不同，现在给出每个地方的坐标，花费的计算方法，以及每个地方可以给哪些地方供水（即对方可以从这里引水），求给所有地方供水的最小花费。

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define maxn 1005
using namespace std;
const int inf=1<<29;

int in[maxn],id[maxn];
int vis[maxn];
int pre[maxn];
int cnt;
struct node
{
    int x,y,z;
}p[maxn*maxn];

int dis(int i,int j)
{
    return abs(p[i].x-p[j].x)+abs(p[i].y-p[j].y)+abs(p[i].z-p[j].z);
}

struct node1
{
    int u,v,w;
}edg[maxn*maxn];

int MST(int root,int V,int E)
{
    int ans=0;
    while(1)
    {
        for(int i=0; i<V; i++) in[i]=inf;
        for(int i=0; i<E; i++)
        {
            int u=edg[i].u;
            int v=edg[i].v;
            if(edg[i].w<in[v]&&u!=v)
            {
                pre[v]=u; in[v]=edg[i].w;
            }
        }
        in[root]=0;
        pre[root]=root;
        for(int i=0; i<V; i++)
        {
            ans+=in[i];
            if(in[i]==inf) return -1;
        }
        memset(vis,-1,sizeof(vis));
        memset(id,-1,sizeof(id));
        cnt=0;
        for(int i=0; i<V; i++)
        {
            int t=i;
            while(vis[t]!=i&&id[t]==-1&&t!=root)
            {
                vis[t]=i;
                t=pre[t];
            }
            if(t!=root&&id[t]==-1)
            {
                for(int u=pre[t]; u!=t; u=pre[u]) id[u]=cnt;
                id[t]=cnt++;
            }
        }
        if(cnt==0) break;
        for(int i=0; i<V; i++)
        {
            if(id[i]==-1) id[i]=cnt++;
        }
        for(int i=0; i<E; i++)
        {
            int u=edg[i].u;
            int v=edg[i].v;
            edg[i].u=id[u];
            edg[i].v=id[v];
            if(id[u]!=id[v]) edg[i].w-=in[v];
        }
        V=cnt;
        root=id[root];
    }
    return ans;
}

int main()
{
    int n,a,b,c;
    while(scanf("%d%d%d%d",&n,&a,&b,&c)!=EOF)
    {
        if(n==0&&a==0&&b==0&&c==0) break;
        int t1=0;
        for(int i=1; i<=n; i++)
        {
            scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].z);
            edg[t1].u=0; edg[t1].v=i;
            edg[t1++].w=a*p[i].z;
        }
        for(int i=1; i<=n; i++)
        {
           int k,v;
           scanf("%d",&k);
           for(int j=0; j<k; j++)
           {
               scanf("%d",&v);
               if(v==i) continue;
               edg[t1].u=i; edg[t1].v=v;
               edg[t1].w=dis(i,v)*b;
               if(p[i].z<p[v].z)
               {
                   edg[t1].w+=c;
               }
               t1++;
           }
        }
        printf("%d
",MST(0,n+1,t1));
    }
    return 0;
}