最近公共祖先LCA(我们的LCA是谁呢)

眉间皱痕 提交于 2019-12-05 09:17:14

P3379 【模板】最近公共祖先(LCA)

例子

LCA(13,14)=12

LCA(13,7)=3

LCA(7,11)=2

LCA(11,2)=2

好了,了解以后我们就来康康怎么求吧

暴力LCA

如果找LCA(6,8),我们就要先让它们的深度相等。此时(deep[1]=0)deep[6]=4,deep[8]=3,我们就先让节点6爬到节点5

然后就是愉快的一起爬

让它们一层一层地向上爬,知道爬到同一节点

但是都知道,一旦这棵树有点高,like this,就会TTT

 

 

 

 

 

 

因为这样爬着太慢了(青蛙爬树都知道跳着爬)

 

 

 

 倍增LCA

所以我们升级了,会跳着爬了

但是怎么升呢,大家都知道,计算机和二进制都离不开关系

所以当然是按照2的次方来跳,不过我们不是从小往大(1,2,4,8,16,32,64...),而是从大到小(...64,,32,16,8,4,2,1)(因为从小到大加着会超过,到时候很麻烦)

所以我们跳的时候就还要计算一下你这么跳会到哪个节点去(不然你乱跳啊)

总所周知,1+1=2(这不是废话吗)

 1 void dfs(int x,int fa)
 2 {
 3     deep[x]=deep[fa]+1;
 4     f[x][0]=fa;
 5     for(int i=1;(1<<i)<=deep[x];i++)
 6     {
 7         f[x][i]=f[f[x][i-1]][i-1];
 8     }
 9     for(int i=0;i<mp[x].size();i++)
10     {
11         if(mp[x][i]==fa)continue;
12         dfs(mp[x][i],x);
13     }
14 }


我们用f[i][j]表示第i个节点向上跳2^j高度后的祖先节点,我们就不用像个憨批一样去搜了(不然你和暴力有什么区别),他就等于2^(j-1)*2也就是向上跳两次2^(j-1)

所以f[i][j]=f[f[i][]j-1][j-1]

所以我们循坏j从小到大,这样就可以用小的来递推大的了

然后搜索的时候顺便把深度也计算了

 

 

预处理完了,我们就要开始干正事了

 1 int LCA(int x,int y)
 2 {
 3     if(deep[x]<deep[y])swap(x,y);
 4     for(int i=log2(deep[x]);i>=0;i--)
 5         if(deep[x]-(1<<i)>=deep[y])
 6             x=f[x][i];
 7     if(x==y)return x;
 8     for(int i=log2(deep[x]);i>=0;i--)
 9     {
10         if(f[x][i]!=f[y][i])
11         {
12             x=f[x][i];
13             y=f[y][i];
14         }
15     }
16     return f[x][0];
17 }

我们先让两个节点到达同一高度(为了方便计算,我们把高的放前面)

然后向上爬

先判断是否在同一点(因为存在两者的祖先是其中一个节点的可能,就是你和你爸的最近公共祖先是你爸)

然后一起向上跳跳跳(jump!!!)

为了避免两者计算出的公共祖先不是最近的(例如你爷爷也是你和你爸的公共祖先)

我们只跳到公共祖先的下一层,也就是第十排的判断

最后再输出它们的最近公共祖先就好了

 1 #include<bits/stdc++.h>
 2 #define N 500005
 3 using namespace std;
 4 int n,m,root;
 5 int f[N][21];
 6 int deep[N];
 7 vector<int> mp[N];
 8 inline int read(){
 9     int x=0,f=1;
10     char ch=getchar();
11     while(ch<'0'||ch>'9'){
12         if(ch=='-')
13             f=-1;
14         ch=getchar();
15     }
16     while(ch>='0'&&ch<='9'){
17         x=(x<<1)+(x<<3)+(ch^48);
18         ch=getchar();
19     }
20     return x*f;
21 }
22 void dfs(int x,int fa)
23 {
24     deep[x]=deep[fa]+1;
25     f[x][0]=fa;
26     for(int i=1;(1<<i)<=deep[x];i++)
27     {
28         f[x][i]=f[f[x][i-1]][i-1];
29     }
30     for(int i=0;i<mp[x].size();i++)
31     {
32         if(mp[x][i]==fa)continue;
33         dfs(mp[x][i],x);
34     }
35 }
36 int LCA(int x,int y)
37 {
38     if(deep[x]<deep[y])swap(x,y);
39     for(int i=log2(deep[x]);i>=0;i--)
40         if(deep[x]-(1<<i)>=deep[y])
41             x=f[x][i];
42     if(x==y)return x;
43     for(int i=log2(deep[x]);i>=0;i--)
44     {
45         if(f[x][i]!=f[y][i])
46         {
47             x=f[x][i];
48             y=f[y][i];
49         }
50     }
51     return f[x][0];
52 }
53 int main()
54 {
55     deep[0]=-1;
56     n=read(),m=read(),root=read();
57     for(int i=1;i<n;i++)
58     {
59         int a,b;
60         a=read(),b=read();
61         mp[a].push_back(b);
62         mp[b].push_back(a);
63     }
64     dfs(root,0);
65     while(m--)
66     {
67         int a,b;
68         a=read(),b=read();
69         printf("%d\n",LCA(a,b));
70     }
71     return 0;
72 }

 

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!