CF629E Famil Door and Roads【树上计数+分类讨论】
Online Judge:Codeforces629E,Luogu-CF629E
Label:树上计数,分类讨论,换根
题目描述
给出一棵n个节点的树。有m个询问,每一个询问包含两个数a、b,我们可以对任意两个不相连的点连一条无向边,并且使得加上这条边后a,b处在一个环内。对于每一个询问,求这样的环的期望长度。
\(2<=n,m<=10^5\)
输入
第一行包括两个整数n,m,分别表示节点数和询问数。
接下来n-1行,每行两个整数u、v表示有一条从u到v的边。
接下来m行,每行两个整数a、b(a≠b),表示一个询问。
输出
对于每一个询问,输出满足条件的环的期望长度。答案保留6位小数。
样例
Input#1
4 3
2 4
4 1
3 2
3 1
2 3
4 1
Output#1
4.00000000
3.00000000
3.00000000
Input#2
3 3
1 2
1 3
1 2
1 3
2 3
Output#2
2.50000000
2.50000000
3.00000000
题解
题目求的是期望,其实就是求两个东西,\(all=\)能形成环的个数,\(ans=\)所有环的长度总和,两者相除得到答案。
转化为树上的计数问题。接下来分两种情况讨论。
先交代下面会用到的数组,及其意义。
\(sz[x]\):以x为根的子树所含节点的个数。
\(dep[x]\):节点深度。
\(fa[x][i=0..17]\):x向上第\(2^i\)个祖先(供后面倍增跳LCA用)。
\(sum[x]\):\(= ∑_{son∈x}dep[son]\),也就是子树中所有点的深度之和。
\(tot[x]\):后面再说。
1.(u,v)不是祖先关系
发现只能将u,v子树里的点连起来。
所以,能形成环的个数\(all=sz[u]*sz[v]\),其中\(sz\)数组表示子树所含节点个数。
所有环的长度之和:
\]
根据图很容易理解。
2.(u,v)是祖先关系
特殊点说明:将深度小的点作为u。son是u的儿子中通向v的那一个。v是深度较大的那一个。
如何求son?倍增向上跳\(dep[v]-dep[u]-1\)步,可以在O(logN)时间内得到。
看下面这幅图,我们只能在v的子树中选一个点,在蓝色部分(除子树son的点)选一个点,将两者相连,才能形成环。
所以,环的个数\(all=sz[v]*(sz[1]-sz[son])\)。
如何求环的长度总和?分成下面几部分分步求解。
这里换个元方便下面表示\(sz1=sz[v]\),\(sz2=(sz[1]-sz[son])\)。\(sz1\)就是粉色部分的点数,\(sz2\)就是蓝色部分的点数。
part1:
\(ans+=(sum[v]-dep[v]*sz1)*sz2+(dep[v]-dep[u])*all\)
part2:
\(ans+=1*all\)
part3:
这个东西与前面几个相比不太好分析。
求树上两点(a,b)的距离就是\(dep[a]+dep[b]-2*dep[LCA(a,b)]\)。
全部加起来,那么现在我们求的就是下面这东西。
\]
前面那个可以O(1)算出。就是后面那一坨如何在可行的时间内弄出?
考虑预处理一个数组\(tot[son]\),表示,若它的父亲节点为u,则\(tot[son]=∑_{x∈bluepart}(dep[x]-2*dep[LCA(u,x)])\)也就是上面式子中后面的部分。
dfs一遍,完成\(tot\)数组的预处理,转移如下。画个图还是很好理解的,当弄到x时,f就是上式的lca。
void dfs2(int x){
int f=fa[x][0];
if(f){
tot[x]=tot[f]+sum[f]-sum[x];
tot[x]-=2ll*(sz[f]-sz[x])*dep[f];
}
for(int i=head[x];i;i=e[i].nxt)if(e[i].to!=fa[x][0])dfs2(e[i].to);
}
综上分两种情况讨论,时间复杂度为\(O(NlogN)\)。
完整代码如下:
//原题CF629E
#include<bits/stdc++.h>
#define int long long
using namespace std;
typedef long long ll;
const int N=1e5+10;
inline int read(){
int x=0;char c=getchar();
while(c<'0'||c>'9')c=getchar();
while(c>='0'&&c<='9')x=(x<<3)+(x<<1)+(c^48),c=getchar();
return x;
}
struct edge{
int to,nxt;
}e[N<<1];
int head[N],ecnt;
inline void link(int u,int v){
e[++ecnt].to=v,e[ecnt].nxt=head[u];
head[u]=ecnt;
}
int n,m;
ll sum[N],tot[N];
int sz[N],dep[N],fa[N][18];
ll gcd(ll a,ll b){return b==0?a:gcd(b,a%b);}
void dfs(int x,int f){
fa[x][0]=f,dep[x]=dep[f]+1,sum[x]=dep[x];
sz[x]=1;
for(int i=head[x];i;i=e[i].nxt){
int y=e[i].to;if(y==f)continue;
dfs(y,x);
sz[x]+=sz[y];
sum[x]+=sum[y];
}
}
void dfs2(int x){
int f=fa[x][0];
if(f){
tot[x]=tot[f]+sum[f]-sum[x];
tot[x]-=2ll*(sz[f]-sz[x])*dep[f];
}
for(int i=head[x];i;i=e[i].nxt)if(e[i].to!=fa[x][0])dfs2(e[i].to);
}
inline int jump(int x,int stp){
for(int i=0;i<=17;i++)if(stp&(1<<i))x=fa[x][i];
return x;
}
inline int LCA(int x,int y){
if(dep[x]<dep[y])swap(x,y);
int stp=dep[x]-dep[y];
x=jump(x,stp);
if(x==y)return x;
for(int i=17;i>=0;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
return fa[x][0];
}
signed main(){
n=read(),m=read();
for(int i=1;i<n;i++){
int u=read(),v=read();
link(u,v),link(v,u);
}
dfs(1,0),dfs2(1);
for(int j=1;j<=17;j++)for(int i=1;i<=n;i++)fa[i][j]=fa[fa[i][j-1]][j-1];
while(m--){
int u=read(),v=read();
if(dep[u]>dep[v])swap(u,v);
int lca=LCA(u,v);
ll all,ans;
if(lca==u){
int son=jump(v,dep[v]-dep[u]-1);
int sz1=sz[v],sz2=sz[1]-sz[son];
all=1ll*sz1*sz2,ans=0;
ans+=1ll*(sum[v]-dep[v]*sz1)*sz2;
ans+=1ll*(tot[son]+sz2*dep[u])*sz1;
ans+=1ll*all*(dep[v]-dep[u]+1);
}
else{
all=1ll*sz[u]*sz[v],ans=0;
ans+=1ll*sz[u]*(sum[v]-dep[v]*sz[v]);
ans+=1ll*sz[v]*(sum[u]-dep[u]*sz[u]);
ans+=1ll*all*(dep[u]+dep[v]-2*dep[lca]+1);
}
//ll g=gcd(all,ans);
//printf("%lld/%lld\n",ans/g,all/g);
printf("%.7f\n",1.0*ans/all);
}
}
CF629E Famil Door and Roads【树上计数+分类讨论】的更多相关文章
- Codeforces Round #343 (Div. 2) E. Famil Door and Roads lca 树形dp
E. Famil Door and Roads 题目连接: http://www.codeforces.com/contest/629/problem/E Description Famil Door ...
- Bzoj4558:分类讨论 计算几何 组合数学
国际惯例的题面: 这题让我爆肝啦......这种计数显然容斥,正好不含任何坏点的我们不会算,但是我们能算至少含零个坏点的,至少含一个坏点的,至少含两个坏点的......所以最终的答案就是(至少含零个坏 ...
- bzoj 3779 重组病毒 好题 LCT+dfn序+线段树分类讨论
题目大意 1.将x到当前根路径上的所有点染成一种新的颜色: 2.将x到当前根路径上的所有点染成一种新的颜色,并且把这个点设为新的根: 3.查询以x为根的子树中所有点权值的平均值. 分析 原题codec ...
- P5979 [PA2014]Druzyny dp 分治 线段树 分类讨论 启发式合并
LINK:Druzyny 这题研究了一下午 终于搞懂了. \(n^2\)的dp很容易得到. 考虑优化.又有大于的限制又有小于的限制这个非常难处理. 不过可以得到在限制人数上界的情况下能转移到的最远端点 ...
- Codeforces 521E - Cycling City(点双连通分量+分类讨论)
Codeforces 题面传送门 & 洛谷题面传送门 大家都是暴力找生成树然后跳路径,代码不到 50 行(暴论)的一说--好,那本蒟蒻决定提供一种代码 150 行,但复杂度也是线性的分类讨论做 ...
- Codeforces 460D Little Victor and Set --分类讨论+构造
题意:从区间[L,R]中选取不多于k个数,使这些数异或和尽量小,输出最小异或和以及选取的那些数. 解法:分类讨论. 设选取k个数. 1. k=4的时候如果区间长度>=4且L是偶数,那么可以构造四 ...
- BZOJ-1067 降雨量 线段树+分类讨论
这道B题,刚的不行,各种碎点及其容易忽略,受不鸟了直接 1067: [SCOI2007]降雨量 Time Limit: 1 Sec Memory Limit: 162 MB Submit: 2859 ...
- UVaLive 6862 Triples (数学+分类讨论)
题意:给定一个n和m,问你x^j + y^j = z^j 的数量有多少个,其中0 <= x <= y <= z <= m, j = 2, 3, 4, ... n. 析:是一个数 ...
- 枚举(分类讨论):BZOJ 1177: [Apio2009]Oil
1177: [Apio2009]Oil Time Limit: 15 Sec Memory Limit: 162 MBSubmit: 1477 Solved: 589[Submit] Descri ...
随机推荐
- CSIC_716_20191104【流程控制语句】
流程控制语句 if 语法结构 if 逻辑判断为真 : xxxxxx else: xxxxx while 语法结构 (continue.break) while 逻辑判断为真: xxxxxxx con ...
- Image 释放
你是用 Image.FromFile取的图片吧,这个方法会锁定图片文件. 有几种方法解决:一:System.Drawing.Image img = System.Drawing.Image.FromF ...
- zabbix配置文件解析
zabbix的配置文件一般有三种:zabbixserver的配置文件zabbix_server.confzabbixproxy的配置文件zabbix_proxy.confzabbix_agentd的配 ...
- day21 生成器,列表解析,三元表达式
Python之路,Day9 = Python基础9 判断可迭代对象和迭代器 from collections import Iterable, Iterator # 导入模块功能,用来判断对象是否为I ...
- Hibernate之Inverse的用法
在多的一端配置Inverse设置为true,来自动管理关系
- mongodb操作指令(二):索引,聚合,管道
索引 索引本质上是树,最小的值在最左边的叶子上,最大的值在最右边的叶子上,使用索引可以提高查询速度(而不用全表扫描),也可以预防脏数据的插入(如唯一索引) 索引通常能够极大的提高查询的效率, 如果没有 ...
- error LNK2001: unresolved external symbol _main解决办法(zz)
error LNK2001: unresolved external symbol _main解决办法 解决外部符号错误:_main,_WinMain@16,__beginthreadex -!t ...
- 自签https证书2(适配新版chrome,不会显示“不安全”)
上一篇博文中介绍了自签https的方法,但是在新版的chrome中会出现这么一个问题:自签ca可以识别,但是证书仍然会判断不安全.为了解决这个问题,博主特地又查了好多资料,终于找到了解决方案. 当然, ...
- QT之QStatusBar
1.QStatusBar一般处于主窗体(QMainWindow)的左下角用于显示比较重要的状态信息.我们通常调用其showMessage()函数 QT官方显示:[slot] void QStatusB ...
- 2day:Python基础
基础知识: 1.python文件的后缀名:.py 2 .Windows Python的执行方式:Python 解释器路径 py文件路径 例:C:\python3\python.exe d:\1.py ...