HDU -- 4009 Transfer water 【无定根的最小树形图 + 思维】

传送门
//题意: 山上有一些村民, 每家可以自己挖井取水, 有一个花费(和高度有关), 也可以选择从比他们高的村民家中引水, 建造管道也有一个花费, 那么问最少的花费是多少.
//思路 : 因为每一个村民家建造水井所需要的花费都是很不确定的, 所以我们最小树形图, 用一个虚根, 而虚边的权值就是每一户打井所需要的花费, 这样以后跑一遍最小树形图就可以出答案.
AC Code

/** @Cain*/
const int maxn=1e3+5;
const int inf=1e9;
int point[maxn][3];
struct node
{
    int u,v,w;
}s[maxn * maxn];

int dist( int i ,int j )
{
    return abs( point[i][0] - point[j][0] ) + abs( point[i][1] - point[j][1] ) + abs( point[i][2] - point[j][2]);
}
int in[maxn],pre[maxn],vis[maxn],id[maxn];

int dirTree(int root,int n,int m)
{
    int res = 0;
    while(1){
        for(int i=0;ifor(int i=0;iint u = s[i].u;
            int v = s[i].v;
            int w = s[i].w;
            if(w < in[v] && u != v){
                in[v] = w;
                pre[v] = u;
            }
        }
        in[root] = 0;
        for(int i=0;iif(in[i] != inf)
                res += in[i];
            else return -1;
        }
        pre[root] = root;
        int newid = 0;
        memset(vis,-1,sizeof(vis));
        memset(id,-1,sizeof(id));
        for(int i=0;iint u = i;
            while(vis[u] != i && id[u] == -1){
                vis[u] = i;
                u = pre[u];
            }
            if(u == root || vis[u] != i) continue;
            for(int v = pre[u] ; v != u ; v = pre[v] )
                id[v] = newid;
            id[u] = newid++;
        }
        if(newid == 0) break;
        for(int i=0;iif(id[i] == -1)
                id[i] = newid++;
        }
        for(int i=0;i.w -= in[s[i].v];
            s[i].u = id[s[i].u];
            s[i].v = id[s[i].v];
        }
        root = id[root];
        n = newid;
    }
    return res;
}
int main()
{
    int n,x,y,z;
    while(~scanf("%d%d%d%d",&n,&x,&y,&z)){
        if(n+x+y+z == 0) break;
        for(int i=1;i<=n;i++){
            scanf("%d%d%d",&point[i][0],&point[i][1],&point[i][2]);
        }
        int cnt = 0;
        for(int i=1;i<=n;i++){   //建造虚边
            s[cnt].u = 0;
            s[cnt].v = i;
            s[cnt++].w = point[i][2] * x;
        }
        for(int i=1;i<=n;i++){
            int k;
            scanf("%d",&k);
            while(k--){
                int h;scanf("%d",&h);
                s[cnt].u = i;
                s[cnt].v = h;
                if(point[h][2] > point[i][2])
                    s[cnt++].w = dist(h,i) * y + z;
                else
                    s[cnt++].w = dist(h,i) * y;
            }
        }
        int res = dirTree(0,n+1,cnt);
        if(res == -1)
            printf("poor XiaoA\n");
        else
            printf("%d\n",res);
    }
}

你可能感兴趣的:(最小树形图)