博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
约会 倍增lca
阅读量:5824 次
发布时间:2019-06-18

本文共 1510 字,大约阅读时间需要 5 分钟。

题意:一棵树,给两个点,求树上有多少点到他俩距离相等

倍增lca,分好多情况讨论。。

#include
#include
#include
#include
#include
#define N 100500using namespace std;int e=1,head[N];struct edge{ int u,v,next;}ed[2*N];void add(int u,int v){ ed[e].u=u;ed[e].v=v; ed[e].next=head[u]; head[u]=e++;}int n,m;int dep[N],fa[N][22],size[N];void dfs(int x){ size[x]=1; for(int i=1;(1<
<=dep[x];i++) fa[x][i]=fa[fa[x][i-1]][i-1]; for(int i=head[x];i;i=ed[i].next){ int v=ed[i].v; if(v==fa[x][0])continue; fa[v][0]=x;dep[v]=dep[x]+1; dfs(v); size[x]+=size[v]; }}int getlca(int x,int y){ if(dep[x]
=dep[y]){ x=fa[x][i]; } if(x==y) return x; for(int i=20;~i;i--) if(fa[x][i]!=fa[y][i]){ x=fa[x][i];y=fa[y][i]; } return fa[x][0];}int find(int x,int y){ int d=dep[x]-y; for(int i=20;~i;i--) if(dep[fa[x][i]]>=d) x=fa[x][i]; return x;}int main(){ scanf("%d",&n);int u,v; for(int i=1;i
dep[v])swap(u,v); int lca=getlca(v,u),mid,len; if(lca==u){ len=dep[v]-dep[u]; if(len%2==1){printf("%d\n",0);continue;} else{ mid=find(v,len/2); printf("%d\n",size[mid]-size[find(v,dep[v]-dep[mid]-1)]); continue; } } len=dep[u]+dep[v]-2*dep[lca]; if(len%2==1){printf("%d\n",0);continue;} else{ mid=find(v,len/2); if(mid==lca){ printf("%d\n",n-size[find(u,dep[u]-dep[lca]-1)]-size[find(v,dep[v]-dep[lca]-1)]); continue; } else{ printf("%d\n",size[mid]-size[find(v,dep[v]-dep[mid]-1)]); continue; } } } return 0;}

转载于:https://www.cnblogs.com/Ren-Ivan/p/7746691.html

你可能感兴趣的文章
iOS 解决UITabelView刷新闪动
查看>>
CentOS 7 装vim遇到的问题和解决方法
查看>>
JavaScript基础教程1-20160612
查看>>
ios xmpp demo
查看>>
python matplotlib 中文显示参数设置
查看>>
【ros】Create a ROS package:package dependencies报错
查看>>
通过容器编排和服务网格来改进Java微服务的可测性
查看>>
re:Invent解读:没想到你是这样的AWS
查看>>
PyTips 0x02 - Python 中的函数式编程
查看>>
使用《Deep Image Prior》来做图像复原
查看>>
Linux基础命令---rmdir
查看>>
Android图片添加水印图片并把图片保存到文件存储
查看>>
开源 免费 java CMS - FreeCMS1.2-标签 infoSign
查看>>
Squid 反向代理服务器配置
查看>>
Java I/O操作
查看>>
Tomcat性能调优
查看>>
Android自学--一篇文章基本掌握所有的常用View组件
查看>>
灰度图像和彩色图像
查看>>
FreeMarker-Built-ins for strings
查看>>
argparse - 命令行选项与参数解析(转)
查看>>