HDU 4009 Transfer water 朱刘算法求无固定根最小树形图

题目链接：HDU 4009

分析：和HDU 2121的做法类似，不同之处在于虚拟根连向所有点的边的权值为原图的点挖井所费的金钱。这样可求得总权值最小的一个（或多个）最小树形图。并且以虚拟根为根求得最小树形图后的原图不一定要求连通。

            （使用虚拟根求无固定根最小树形图的方法详见http://blog.csdn.net/vikotse/article/details/23036055）

代码如下：

#include <cstdio>
#include <cstdlib>
#include <cstring>

typedef long long ll;
const int N = 1111;
const int M = 1111111;
const int inf = (int)1e9;
int pre[N], ID[N], vis[N], In[N], tot, X, Y, Z;

struct Point
{
    int x, y, z;
} P[N];

struct Edge
{
    int u, v, cost;
} E[M];

void addedge(int a, int b, int c) { E[tot].u = a, E[tot].v = b, E[tot++].cost = c; }

int dist(int a, int b)
{
    int mht = abs(P[a].x - P[b].x) + abs(P[a].y - P[b].y) + abs(P[a].z - P[b].z);
    if(P[b].z >= P[a].z) return mht * Y;
    else return mht * Y + Z;
}

ll Directed_MST(int root, int NV, int NE)
{
    ll ret = 0;
    while(true) {
        // 1.找最小入边
        for(int i = 0; i < NV; i++) In[i] = inf;
        for(int i = 0; i < NE; i++) {
            int u = E[i].u;
            int v = E[i].v;
            if(E[i].cost < In[v] && u != v) {
                pre[v] = u;
                In[v] = E[i].cost;
            }
        } // 为之后的遍历做准备，in[]用于存储最小边集
        for(int i = 0; i < NV; i++) {
            if(i == root) continue;
            if(In[i] == inf) return -1; // 除了跟以外有点没有入边,则根无法到达它
        }
        // 2.找环
        int cntnode = 0;
        memset(ID, -1, sizeof(ID));
        memset(vis, -1, sizeof(vis));
        In[root] = 0;
        for(int i = 0; i < NV; i++) { // 标记每个环
            ret += In[i];
            int v = i;
            while(vis[v] != i && ID[v] == -1 && v != root) {
                vis[v] = i;
                v = pre[v];
            }
            if(v != root && ID[v] == -1) {
                for(int u = pre[v] ; u != v ; u = pre[u]) {
                    ID[u] = cntnode;    // 给环上的结点赋值一个统一的新编号
                }
                ID[v] = cntnode++;
            }
        }
        if(cntnode == 0) break; // 无环
        for(int i = 0; i < NV; i++) {
            if(ID[i] == -1) ID[i] = cntnode++;  // 给无环的结点重新编号
        }
        // 3.缩点,重新标记
        for(int i = 0; i < NE; i++) {
            int v = E[i].v;
            E[i].u = ID[E[i].u];
            E[i].v = ID[E[i].v];
            if(E[i].u != E[i].v) {
                E[i].cost -= In[v];
            }
        }
        NV = cntnode;
        root = ID[root]; // 给根新的标号
    }
    return ret;
}

int main()
{
    // freopen("in", "r", stdin);
    int n, k, to;
    while(scanf("%d%d%d%d", &n, &X, &Y, &Z), n || X || Y || Z) {
        tot = 0;
        for(int i = 0; i < n; i++) scanf("%d%d%d", &P[i].x, &P[i].y, &P[i].z);
        for(int i = 0; i < n; i++) {
            scanf("%d", &k);
            while(k--) {
                scanf("%d", &to);
                addedge(i, to - 1, dist(to - 1, i));
            }
        }
        for(int i = 0; i < n; i++) addedge(n, i, P[i].z * X);
        ll as = Directed_MST(n, n + 1, tot);
        if(as == -1) puts("poor XiaoA");
        else printf("%I64d\n", as);
    }
    return 0;
}