LCA 各种神奇的LCA优化方法
LCA(Least Common Ancestors)
树上问题的一种。
朴素lca很简单啦,我就不多说了,时间复杂度n^2
1.倍增LCA
时间复杂度 nlongn+klogn
其实是一种基于朴素lca的优化方法,
朴素lca只能一层层的向上查询,而这个有一定状态压缩的想法
即每一次跳2^i层,让O(n)的查找变成O(logn)。
以上就是我对倍增lca的理解。
以洛谷P3128为例,
[USACO15DEC]最大流Max Flow
题目描述
Farmer John has installed a new system of N−1 pipes to transport milk between the N stalls in his barn (2≤N≤50,000), conveniently numbered 1…N. Each pipe connects a pair of stalls, and all stalls are connected to each-other via paths of pipes.
FJ is pumping milk between K pairs of stalls (1≤K≤100,000). For the ith such pair, you are told two stalls si and ti, endpoints of a path along which milk is being pumped at a unit rate. FJ is concerned that some stalls might end up overwhelmed with all the milk being pumped through them, since a stall can serve as a waypoint along many of the Kpaths along which milk is being pumped. Please help him determine the maximum amount of milk being pumped through any stall. If milk is being pumped along a path from si to ti, then it counts as being pumped through the endpoint stalls si and
ti, as well as through every stall along the path between them.
FJ给他的牛棚的N(2≤N≤50,000)个隔间之间安装了N-1根管道,隔间编号从1到N。所有隔间都被管道连通了。
FJ有K(1≤K≤100,000)条运输牛奶的路线,第i条路线从隔间si运输到隔间ti。一条运输路线会给它的两个端点处的隔间以及中间途径的所有隔间带来一个单位的运输压力,你需要计算压力最大的隔间的压力是多少。
输入输出格式
输入格式:
The first line of the input contains N and K.
The next N−1 lines each contain two integers x and y (x≠y) describing a pipe
between stalls x and y.
The next K lines each contain two integers s and t describing the endpoint
stalls of a path through which milk is being pumped.
输出格式:
An integer specifying the maximum amount of milk pumped through any stall in the
barn.
差分数组+倍增LCA
先DFS建树,再倍增求lca,最后在求lca时*护一个sum数组(表示被遍历的次数),最后再一遍dfs求出最大被遍历次数
#include <cstdio> #include <cmath> #include <cstring> #include <algorithm> #include <iostream> #include <queue> using namespace std; #define N 100001 int n,K; struct node { int to,next; }e[N]; int ans; int dep[N],fa[N][32],head[N],cnt,sum[N]; void add(int x,int y) { e[cnt].to=y; e[cnt].next=head[x]; head[x]=cnt++; return ; } void init() { memset(head,-1,sizeof(head)); return ; } void dfs(int x,int from) { dep[x]=dep[from]+1; fa[x][0]=from; for(int i=head[x];i!=-1;i=e[i].next) { if(e[i].to!=from) { dfs(e[i].to,x); } } return ; } int lca(int x,int y) { if(dep[x]<dep[y]) { swap(x,y); } int dep1=dep[x]-dep[y]; for(int i=0;i<=20;i++) { if((dep1&(1<<i))!=0) { x=fa[x][i]; } } if(x==y) { return x; } for(int i=20;i>=0;i--) { if(fa[x][i]!=fa[y][i]) { x=fa[x][i]; y=fa[y][i]; } } return fa[x][0]; } void renew(int x) { for(int i=head[x];i!=-1;i=e[i].next) { int to1=e[i].to; if(to1!=fa[x][0]) { renew(to1); sum[x]+=sum[to1]; } } ans=max(ans,sum[x]); return ; } int main() { init(); scanf("%d%d",&n,&K); for(int i=1;i<n;i++) { int x,y; scanf("%d%d",&x,&y); add(x,y); add(y,x); } dep[1]=1; dfs(1,0); for(int i=1;i<=20;i++) { for(int j=1;j<=n;j++) { fa[j][i]=fa[fa[j][i-1]][i-1]; } } for(int i=1;i<=K;i++) { int x,y; scanf("%d%d",&x,&y); int z=lca(x,y); sum[z]--; sum[fa[z][0]]--; sum[x]++; sum[y]++; } renew(1); printf("%d\n",ans); return 0; }
明天继续更新,今天先写到这里...
2.离线tarjan_lca
时间复杂度O(n)
唯一一个lcaO(n)的优化方法...
但是实用性低,代码难写,思想复杂...
并查集+树形DP+离线思想
把查询离线,一遍dfs遍历,查出lca并储存
例题 JLOI2014松鼠的新家
题解 查分数组+tarjanlca
JLOI2014松鼠的新家
Description
松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房间,并且有n-1根树枝连接,每个房间都可以相互到达,且俩个房间之间的路线都是唯一的。天哪,他居然真的住在“树”上。松鼠想邀请****前来参观,并且还指定一份参观指南,他希望**能够按照他的指南顺序,先去a1,再去a2,……,最后到an,去参观新家。
可是这样会导致**重复走很多房间,懒惰的**不听地推辞。可是松鼠告诉他,每走到一个房间,他就可以从房间拿一块糖果吃。**是个馋家伙,立马就答应了。
现在松鼠希望知道为了保证**有糖果吃,他需要在每一个房间各放至少多少个糖果。因为松鼠参观指南上的最后一个房间an是餐厅,餐厅里他准备了丰盛的大餐,所以当**在参观的最后到达餐厅时就不需要再拿糖果吃了。
Input
第一行一个整数n,表示房间个数
第二行n个整数,依次描述a1-an
接下来n-1行,每行两个整数x,y,表示标号x和y的两个房间之间有树枝相连。
Output
一共n行,第i行输出标号为i的房间至少需要放多少个糖果,才能让**有糖果吃。
Sample Input
Sample Output
HINT
30%的数据,n<=4000
80%的数据,n<=50000
100%的数据,2<= n <=300000
3.树链剖分_lca
树剖的用处实在很多
时间复杂度:O(nlogn)
定义:
树链:就是树上的路径。
剖分:就是把路径分类为重链和轻链。
树链剖分:把一棵树剖分为若干条链,然后利用数据结构(树状数组,SBT,Splay,线段树等等)去*护每一条链,复杂度为O(logn)
重儿子:siz[v]为u的子节点中siz值最大的,那么v就是u的重儿子。
轻儿子:u的其它子节点。
重边:点u与其重儿子的连边。
轻边:点u与其轻儿子的连边。
重链:由重边连成的路径。
轻链:轻边。剖分后的树有如下性质:
性质1:如果(v,u)为轻边,则siz[v] * 2 <= siz[u];性质2:从根到某一点的路径上轻链、重链的个数都不大于logn。
那么,树链剖分的第一步当然是对树进行轻重边的划分。剖分过程分为两次dfs,或者bfs也可以。
如果是两次dfs,那么第一次dfs就是找重边,也就是记录下所有的重边。
然后第二次dfs就是连接重边形成重链,具体过程就是:以根节点为起点,沿着重边向下拓展,拉成重链,不在当前重链上的节点,都以该节点为起点向下重新拉一条重链。
剖分完毕后,每条重链相当于一段区间,然后用数据结构去*护,把所有重链首尾相接,放到数据结构上,然后*护整体。
在这里,当然有很多数组,现在我来分别介绍它们的作用:
size1[]数组,用来保存以x为根的子树节点个数
anc[]数组,用来保存当前节点的所在链的顶端节点
son[]数组,用来保存重儿子
dep[]数组,用来保存当前节点的深度
fa[]数组,用来保存当前节点的父亲
Nearest Common Ancestors
Description
给定N个节点的一棵树,有K次查询,每次查询a和b的最近公共祖先。
样例中的16和7的公共祖先(LCA:Least Common Ancestors)是4。
Input
第一行两个整数N(1 < N <= 105)、K(1 <= K <= 105)
第2~N行,每行两个整数a、b(1 <= a,b <= N),表示a是b的父亲。
第N+1~N+K+1行,每行两个整数a、b(1 <= a,b <= N),表示询问a和b的最近公共祖先是谁。
Output
输出K行,第i行表示第i个查询的最近公共祖先是谁。
Sample Input
Sample Output
#include <cstdio> #include <cstring> #include <cmath> #include <algorithm> #include <queue> using namespace std; #define N 100005 int n,K; struct node { int next,to; }e[2*N]; int head[N],dep[N],anc[N],son[N],fa[N]; int in1[N],cnt,size1[N],root; void init() { memset(head,-1,sizeof(head)); return ; } void add(int x,int y) { e[cnt].to=y; e[cnt].next=head[x]; head[x]=cnt++; return ; } void dfs1(int x,int from) { fa[x]=from; size1[x]=1; for(int i=head[x];i!=-1;i=e[i].next) { int to1=e[i].to; if(to1!=from) { dep[to1]=dep[x]+1; dfs1(to1,x); size1[x]+=size1[to1]; if(size1[to1]>size1[son[x]]) { son[x]=to1; } } } return ; } void dfs2(int x,int top1) { anc[x]=top1; if(son[x]) { dfs2(son[x],top1); } for(int i=head[x];i!=-1;i=e[i].next) { int to1=e[i].to; if(to1!=fa[x]&&to1!=son[x]) { dfs2(to1,to1); } } return ; } int getlca(int x,int y) { int ancx,ancy; while(1) { ancx=anc[x],ancy=anc[y]; if(ancx==ancy)break; if(dep[ancx]<dep[ancy]) { y=fa[ancy]; }else { x=fa[ancx]; } } if(dep[x]<dep[y]) { return x; } return y; } int main() { init(); scanf("%d%d",&n,&K); for(int i=1;i<n;i++) { int x,y; scanf("%d%d",&x,&y); add(x,y); add(y,x); in1[y]++; } for(int i=1;i<=n;i++) { if(in1[i]==0) { root=i; dfs1(i,0); } } dfs2(root,root); for(int i=1;i<=K;i++) { int x,y; scanf("%d%d",&x,&y); printf("%d\n",getlca(x,y)); } return 0; }
4.lca转RMQ
依据DFS序,用RMQ优化求区间最*值(dep的值)
而区间最*值即为所求
样题同上....
#include <cstdio> #include <algorithm> #include <cstring> #include <cmath> #include <iostream> #include <queue> using namespace std; #define N 200005 int n,m,in1[N],a[N*2]; struct node { int to,next; }e[N]; int head[N],place[N*2][20],dep[N*2],minn[N*2][20],cnt,num; void init() { memset(head,-1,sizeof(head)); return ; } void add(int x,int y) { e[cnt].to=y; e[cnt].next=head[x]; head[x]=cnt++; return ; } void dfs(int from,int x) { a[x]=++num; place[num][0]=x; dep[num]=dep[a[from]]+1; minn[num][0]=dep[num]; for(int i=head[x];i!=-1;i=e[i].next) { int to1=e[i].to; if(to1!=from) { dfs(x,to1); a[x]=++num; dep[num]=dep[a[from]]+1; minn[num][0]=dep[num]; place[num][0]=x; } } return ; } int main() { init(); scanf("%d%d",&n,&m); for(int i=1;i<n;i++) { int x,y; scanf("%d%d",&x,&y); add(x,y); add(y,x); in1[y]++; } for(int i=1;i<=n;i++) { if(in1[i]==0) { dfs(0,i); break; } } for(int j=1;j<=17;j++) { for(int i=1;i<=2*n-1;i++) { if(i+(1<<j-1)>n*2-1)break; if(minn[i][j-1]<minn[i+(1<<(j-1))][j-1]) { minn[i][j]=minn[i][j-1]; place[i][j]=place[i][j-1]; }else { minn[i][j]=minn[i+(1<<(j-1))][j-1]; place[i][j]=place[i+(1<<(j-1))][j-1]; } } } for(int i=1;i<=m;i++) { int x,y; scanf("%d%d",&x,&y); if(a[x]>a[y]) { swap(x,y); } int len=a[y]-a[x]+1; int t=(int)log2(len); if(minn[a[x]][t]<minn[a[y]-(1<<t)+1][t]) { printf("%d\n",place[a[x]][t]); }else { printf("%d\n",place[a[y]-(1<<t)+1][t]); } } return 0; }
膜拜众大佬,如有问题,请帮我指出,O(∩_∩)O谢谢!