hdu 4009 最小树形图

思路：唯一一个值得一提的就是建一个0号根节点，往每个房子建一条边，权值为房子的高度乘以X。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define Maxn 1010
#define Maxm Maxn*Maxn
#define inf 100000000
using namespace std;
int head[Maxn],vi[Maxn],id[Maxn],in[Maxn],pre[Maxn],cnt,X,Y,Z,n,e;
struct Edge{
    int u,v,next,val;
}edge[Maxm];
struct Point{
    int x,y,z;
}p[Maxn];
void init()
{
    memset(head,-1,sizeof(head));
    e=0;
}
void add(int u,int v,int val)
{
    edge[e].u=u,edge[e].v=v,edge[e].val=val,edge[e].next=head[u],head[u]=e++;
}
int Dis(Point a,Point b)
{
    return abs(a.x-b.x)+abs(a.y-b.y)+abs(a.z-b.z);
}
int directed_MST(int root,int N,int E)
{
    int ans=0,i,j,u,v;
    while(1)
    {
        for(i=0;i<N;i++)
            in[i]=inf;
        for(i=0;i<E;i++)
        {
            u=edge[i].u,v=edge[i].v;
            if(in[v]>edge[i].val&&u!=v)
            {
                pre[v]=u;
                in[v]=edge[i].val;
            }
        }
        for(i=0;i<N;i++)
        {
            if(i==root) continue;
            if(in[i]==inf) return -1;
        }
        memset(vi,-1,sizeof(vi));
        memset(id,-1,sizeof(id));
        cnt=0;
        in[root]=0;
        for(i=0;i<N;i++)//枚举每个可能是环上的点
        {
            ans+=in[i];
            v=i;
            while(vi[v]!=i&&id[v]==-1&&v!=root)
            {
                vi[v]=i;v=pre[v];
            }
            if(v!=root&&id[v]==-1)//如果退出上个循环的条件不是v==root，就是vi[i]==i，即找到了环
            {
                for(u=pre[v];u!=v;u=pre[u])//对环进行统一标号
                    id[u]=cnt;
                    id[v]=cnt++;
            }
        }
        if(cnt==0) break;//无环 ，退出
        for(i=0;i<N;i++) if(id[i]==-1) id[i]=cnt++;
        for(i=0;i<E;i++)//进行缩点，并修改每条边的权值。
        {
            v=edge[i].v;
            edge[i].u=id[edge[i].u];
            edge[i].v=id[edge[i].v];
            if(edge[i].u!=edge[i].v)
                edge[i].val-=in[v];
        }
        N=cnt;
        root=id[root];
    }
    return ans;
}
int main()
{
    int i,j,a,b,c,k;
    while(scanf("%d%d%d%d",&n,&X,&Y,&Z),n|X|Y|Z)
    {
        init();
        for(i=1;i<=n;i++)
            scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].z);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&k);
            for(j=1;j<=k;j++)
            {
                scanf("%d",&a);
                if(a==i) continue;
                if(p[a].z>p[i].z)
                    add(i,a,Dis(p[a],p[i])*Y+Z);
                else
                    add(i,a,Dis(p[a],p[i])*Y);
            }
        }
        for(i=1;i<=n;i++)
            add(0,i,X*p[i].z);
        printf("%d\n",directed_MST(0,n+1,e));
    }
    return 0;
}