首页 > 代码库 > codeforces E. Famil Door and Roads 期望
codeforces E. Famil Door and Roads 期望
一棵树,n个节点,边长为1,有q个询问,每个询问给出u,v(u != v),问在树上等概率加一条边,如果使得u,v在一个环内,则这种加边方式是合法的,此时的值为环的长度,所有合法的加边方式出现的概率相等,问值的期望。
2 <= n,m <= 10^5
对于u,v原来路径上的边一定在环内,贡献为1,新加的边也一定在环内,贡献为1,求其余的边的贡献就行了
分2种情况考虑:
1.lca(u,v) 不等于u 和 v
2.lca(u,v) 为u 或者 v
代码:
//File Name: cf629E.cpp //Created Time: 2017年01月05日 星期四 17时20分29秒 #include <bits/stdc++.h> #define LL long long using namespace std; const int MAXN = 100000 + 2; int siz[MAXN],dis[MAXN]; int pa[MAXN][20]; LL f[MAXN],g[MAXN]; vector<int> G[MAXN]; void dfs0(int u,int p){ siz[u] = 1; dis[u] = dis[p] + 1; f[u] = 0; for(int i=0;i<G[u].size();++i){ int v = G[u][i]; if(v == p) continue; pa[v][0] = u; dfs0(v,u); siz[u] += siz[v]; f[u] += f[v] + siz[v]; } } void dfs1(int u,int p,int n){ for(int i=0;i<G[u].size();++i){ int v = G[u][i]; if(v == p) continue; g[v] = n + g[u] - 2 * siz[v]; dfs1(v,u,n); } } void cal_pa(int n){ for(int j=1;(1<<j)<=n;++j){ for(int i=1;i<=n;++i){ if(pa[i][j-1] != -1) pa[i][j] = pa[pa[i][j-1]][j-1]; } } } int cal_lca(int a,int b){ if(dis[a] < dis[b]) swap(a,b); int cnt = 0; for(;(1<<cnt)<=dis[a];++cnt); --cnt; for(int j=cnt;j>=0;--j){ if(dis[a] - (1<<j) >= dis[b]) a = pa[a][j]; } if(a == b) return a; for(int j=cnt;j>=0;--j){ if(pa[a][j] != -1 && pa[a][j] != pa[b][j]) a = pa[a][j],b = pa[b][j]; } return pa[a][0]; } int cal(int u,int v){ int cnt = 0; for(;(1<<cnt)<=dis[u];++cnt); --cnt; for(int j=cnt;j>=0;--j){ if(dis[u] - (1<<j) > dis[v]) u = pa[u][j]; } return u; } void solve(int n,int m){ memset(pa,-1,sizeof(pa)); dfs0(1,0); //dis,siz,in,f,pa[i][0],dep g[1] = f[1]; dfs1(1,0,n); //g cal_pa(n); // pa int u,v,lca,w; while(m--){ scanf("%d %d",&u,&v); lca = cal_lca(u,v); if(lca != u && lca != v){ int tmp = dis[u] + dis[v] - 2 * dis[lca] + 1; double ans = tmp + (f[u] + 0.0) / siz[u] + (f[v] + 0.0) / siz[v]; printf("%.15f\n",ans); } else{ if(lca == u) swap(u,v); w = cal(u,v); if(pa[w][0] != v){ printf("-1"); return ; } int tmp = dis[u] - dis[v] + 1; double ans = tmp + (f[u] + 0.0) / siz[u] + (g[v] - f[w] - siz[w] + 0.0) / (n - siz[w]); printf("%.15f\n",ans); } } } int main(){ int n,m; scanf("%d %d",&n,&m); for(int i=1,u,v;i<n;++i){ scanf("%d %d",&u,&v); G[u].push_back(v); G[v].push_back(u); } solve(n,m); return 0; }
codeforces E. Famil Door and Roads 期望
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。