（最小树形图）POJ 3164

题意：

给定一个有向图，求以某个给定顶点为根的有向生成树（也就是说沿着这N-1条有向边可以从根走到任一点），使权和最小。

分析：

这题直接朱刘算法，虽然看起来很偏门，但是还是要学一下。算法过程很多博客都有了，这里就不赘述了。

不过貌似CSU1828是朱刘算法+AC自动机，可以刷一刷。

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
#include <cmath>

using namespace std;

const int inf = 0x3f3f3f3f;
const int maxn = 110;


int n, m;
double x[maxn], y[maxn];

double dis(double x1, double y1, double x2, double y2) {
    return sqrt(pow(x1 - x2, 2) + pow(y1 - y2, 2));
}

struct Edge {
    int u, v;
    double c;
};

Edge edge[10010];

double in[maxn];
int pre[maxn];
int id[maxn];
int vis[maxn];

double MTV(int root) {
    double ans = 0;
    while(1) {
        for(int i = 0; i < n; i++) in[i] = inf;
        for(int i = 0; i < m; i++) {
            int u = edge[i].u;
            int v = edge[i].v;
            if(edge[i].c < in[v] && u != v) {
                in[v] = edge[i].c;
                pre[v] = u;
            }
        }

        for(int i = 0; i < n; i++) {
            if(i == root) continue;
            if(in[i] == inf) return -1;
        }

        int cnt = 0;
        memset(id, -1, sizeof(id));
        memset(vis, -1, sizeof(vis));
        in[root] = 0;
        for(int i = 0; i < n; i++) {
            ans += 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] = cnt;
                }
                id[v] = cnt++;
            }
        }
        if(cnt == 0) break;

        for(int i = 0; i < n; i++) {
            if(id[i] == -1) id[i] = cnt++;
        }

        for(int i = 0; i < m; i++) {
            int v = edge[i].v;
            edge[i].v = id[edge[i].v];
            edge[i].u = id[edge[i].u];
            if(edge[i].u != edge[i].v) {
                edge[i].c -= in[v];
            }
        }
        n = cnt;
        root = id[root];
    }
    return ans;
}



int main() {
    while(~scanf("%d%d", &n, &m)) {
        for(int i = 0; i < n; i++) {
            scanf("%lf%lf", &x[i], &y[i]);
        }
        for(int i = 0; i < m; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            u--, v--;
            if(u == v)edge[i] = Edge{u, v, inf};
            else edge[i] = Edge{u, v, dis(x[u], y[u], x[v], y[v])};
        }
        double ans = MTV(0);
        if(ans < 0)puts("poor snoopy");
        else printf("%.2f
", ans);
    }

    return 0;

}