题目描述
富爷说来一棵树,于是大头栽了一棵树。树大了,有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);
}
}