11年福州 F 次小生成树变形 树形DP

单独练习的时候，做到最小生成树处理出来后，就不知道怎么处理了。关键是两个子树间的距离，不能在有效的时间复杂度内解决。
实际上，对于一个最小生成树，两个子树间距离可以用dp来做。
假设从root开始向下搜索，那么dp[root][u]表示root到u和以u为根节点的子树的最小距离，此处距离即是取的不在最小生成树中的原图的边。那么很容易发现是一个简单的树形DP过程。
然后，如果对于两个子树呢？两个子树是的并集是最小生成树，设两个子树的连接边为(u,v)，那么此时两个子树最小距离为dp[k][v]，k为u或者以u为根节点的子节点。
实际上，这种方法应该可以扩展到最小生成树的一些其他题目？

调代码的时候出现一些小插曲，prime算法携程dijkstra一直没改出来……prime算法加新点进去后，dis更新为dis[v] = min(dis[v], g[u][v])，Dijkstra为dis[v] = min(dis[v], dis[u] + g[u][v])

失分点/学习地方：
1. 两个子树距离
2. 最小生成树prime算法

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
#define inf (1000000007)
#define LL long long
const int MAXN = 3000 + 5;
int n, m;
int g[MAXN][MAXN], dp[MAXN][MAXN], best[MAXN][MAXN];
LL mst;
vector<int>e[MAXN];
///prime use
int vis[MAXN], dis[MAXN], pre[MAXN];
void prime()
{
    for(int i = 0 ; i < n ; i++) vis[i] = 0, dis[i] = inf, pre[i] = -1;
    dis[0] = 0;
    vis[0] = 1;
    for(int i = 1 ; i < n ; i++) {
        if(dis[i] > dis[0] + g[0][i]) {
            dis[i] = dis[0] + g[0][i];
            pre[i] = 0;
        }
    }
    mst = 0;
    for(int T = 2 ; T <= n ; T++) {
        int mindis = inf, re = -1;
        for(int i = 0 ; i < n ; i++) {
            if(vis[i] == 0 && dis[i] < mindis) {
                mindis = dis[i];
                re = i;
            }
        }

        mst += mindis;
        vis[re] = 1;
        if(pre[re] != -1) {
            e[pre[re]].push_back(re);
            e[re].push_back(pre[re]);
        }
        for(int i = 0 ; i < n ; i++) {
            if(vis[i] == 0 && dis[i] >  g[re][i]) {
                dis[i] = g[re][i];
                pre[i] = re;
            }
        }
    }
}
int dfs1(int u, int fa, int root)
{
//    printf("u = %d, fa = %d, root = %d\n", u, fa, root);
    for(int i = 0 ; i < (int)e[u].size() ; i++) {
        int v = e[u][i];
        if(v == fa) continue;
        dp[root][u] = min(dp[root][u], dfs1(v, u, root));
    }
    if(fa != root) dp[root][u] = min(dp[root][u], g[root][u]);
    return dp[root][u];
}
int dfs2(int u, int fa, int root)
{
    int ans = dp[u][root];
    for(int i = 0 ; i < (int)e[u].size() ; i++) {
        int v = e[u][i];
        if(v == fa) continue;
        ans = min(ans, dfs2(v, u, root));
    }
    return ans;
}
int main()
{
    while(scanf("%d%d", &n, &m) != EOF && n + m) {
        for(int i = 0 ; i <= n ; i++) for(int j = 0 ; j <= n ; j++) g[i][j] = dp[i][j] = best[i][j] = inf;
        for(int i = 0 ; i <= n ; i++) e[i].clear();
        for(int i = 1 ; i <= m ; i++) {
            int u, v, w; scanf("%d%d%d", &u, &v, &w);
            g[u][v] = g[v][u] = w;
        }
        prime();
//        printf("mst = %d\n", mst);
//        puts("pre");
//        for(int i = 0 ; i < n ; i++) printf("%d ", pre[i]);
//        puts("\npre");
        for(int i = 0 ; i < n ; i++) dfs1(i, -1, i);
        for(int i = 0 ; i < n ; i++) {
            for(int j = 0 ; j < (int)e[i].size() ; j++){
                int v = e[i][j];
                best[v][i] = dfs2(v, i, i);
            }
        }
//        printf("\ndp\n");
//        for(int i = 0 ; i < n ; i++) {
//            for(int j = 0 ; j < n ; j++) printf("%d ", dp[i][j]);
//            printf("\n");
//        }
//        puts("dp");
//        printf("best\n");
//        for(int i = 0 ; i < n ; i++) {
//            for(int j = 0 ; j < n ; j++) printf("%d ", best[i][j]);
//            printf("\n");
//        }
//        puts("best");
        double res = 0;
        int q;  scanf("%d", &q);
        for(int i = 0 ; i < q ; i++) {
            int u, v, w;    scanf("%d%d%d", &u, &v, &w);
            int ok = 0;
            for(int j = 0 ; j < (int)e[u].size() ; j++) {
                if(e[u][j] == v) {
                    ok = 1;
                    break;
                }
            }
//            printf("ok = %d, g[u][v] = %d\n", ok, g[u][v]);
            if(ok == 0) res += mst;
            else {
                double res1 = 1.0 * mst - g[u][v] + w;
                double res2 = 1.0 * mst - g[u][v] + best[u][v];
//                printf("res1 = %d, res2 = %d\n", res1, res2);
                res += min(res1, res2);
            }
        }
        printf("%.4f\n", res / (1.0 * q));
    }
    return 0;
}