富爷说是一颗树

题目描述

富爷说来一棵树，于是大头栽了一棵树。树大了，有n个点和n - 1条边，任意两个点都是联通的，点的标号为1 - n。爱树的大头和富爷在树上安居乐业，但大头住在u，而富爷住在v,他们都很不高兴，因为u到v有且只有一条简单路径。
当然了，树王富爷找到了解决办法，他打算带着大头再给树建一条边（保证不是自环），而且他们会在n * (n - 1) / 2的方案中随机选择一种。
但，要让富爷和大头开心是有条件的。只有新建边之后，富爷去大头家以及大头去富爷家存在两条路径不会走相同的边时，他们才会呵呵（也就是说 存在一个简单环包含u和v）。
不开心的事情选择忘记。当富爷和大头开心时，你能得到愉快值等于环的大小。所以，你要告诉富爷和大头，当他们开心时（只考虑在环内），他们的期望愉悦值。

倍增一下

我们维护每个节点的size以及f、g
f[x]表示x到其子树内所有节点的距离和。
g[x]表示x的父亲到除了x子树内的所有节点的距离和。
两遍dfs处理f、g。
然后询问u,v，如果u和v不是祖孙关系，那么可行点对一个在u子树内一个在v子树内。如果v是u祖先，设w是u祖先，且v是w父亲，那么可行点对一个在u子树内，另一个不在w子树内。
通过f和g可以统计答案。

#include<cstdio>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
typedef double db;
const int maxn=100000+10;
ll f[maxn],g[maxn],cnt;
int fa[maxn][25],size[maxn],d[maxn];
int h[maxn],go[maxn*2],next[maxn*2];
int i,j,k,l,t,n,m,u,v,w,tot;
db ans;
int read(){
    int x=0;
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x;
}
void add(int x,int y){
    go[++tot]=y;
    next[tot]=h[x];
    h[x]=tot;
}
void dfs(int x,int y){
    fa[x][0]=y;
    d[x]=d[y]+1;
    size[x]=1;
    int t=h[x];
    while (t){
        if (go[t]!=y){
            dfs(go[t],x);
            size[x]+=size[go[t]];
            f[x]+=(ll)(size[go[t]]+f[go[t]]);
        }
        t=next[t];
    }
}
void dg(int x,int y){
    if (x!=1) g[x]=f[y]-(ll)size[x]-f[x]+g[y]+(ll)(n-size[y]);
    int t=h[x];
    while (t){
        if (go[t]!=y) dg(go[t],x);
        t=next[t];
    }
}
int lca(int x,int y){
    if (d[x]<d[y]) swap(x,y);
    if (d[x]!=d[y]){
        int j=floor(log(n)/log(2));
        while (j>=0){
            if (d[fa[x][j]]>=d[y]) x=fa[x][j];
            j--;
        }
    }
    if (x==y) return x;
    int j=floor(log(n)/log(2));
    while (j>=0){
        if (fa[x][j]!=fa[y][j]){
            x=fa[x][j];
            y=fa[y][j];
        }
        j--;
    }
    return fa[x][0];
}
int get(int x,int y){
    int j=floor(log(n)/log(2));
    while (j>=0){
        if (d[fa[x][j]]>d[y]) x=fa[x][j];
        j--;
    }
    return x;
}
int main(){
    //freopen("t1.in","r",stdin);freopen("t1.out","w",stdout);
    n=read();m=read();
    fo(i,1,n-1){
        j=read();k=read();
        add(j,k);add(k,j);
    }
    dfs(1,0);
    fo(j,1,floor(log(n)/log(2)))
        fo(i,1,n)
            fa[i][j]=fa[fa[i][j-1]][j-1];
    dg(1,0);
    while (m--){
        u=read();v=read();
        if (d[u]<d[v]) swap(u,v);
        w=lca(u,v);
        if (w==v){
            w=get(u,v);
            ans=(db)(d[u]-d[v])*size[u]*(n-size[w]);
            ans+=(db)size[u]*(n-size[w]);
            ans+=(db)f[u]*(n-size[w]);
            ans+=(db)g[w]*size[u];
            cnt=(ll)size[u]*(n-size[w]);
        }
        else{
            ans=(db)(d[u]+d[v]-d[w]*2)*size[u]*size[v];
            ans+=(db)size[u]*size[v];
            ans+=(db)f[u]*size[v];
            ans+=(db)f[v]*size[u];
            cnt=(ll)size[u]*size[v];
        }
        printf("%.6lf\n",ans/cnt);
    }
}