Problem M: [模板]最近公共祖先LCA
Time Limit: 3 Sec Memory Limit: 128 MB
Description
给出N,Q
代表一个树有N个点 ,树的根为1
Q代表有Q个询问,询问A,B的最近公共祖先是哪一个.
Input
第一行给出N,Q
下面N-1行描述这个树,格式为A,B,代表A为B的Son
现来下Q行
给出A,B,代表进行询问A,B的LCA是哪一个.
Output
针对每个询问输出结果
Sample Input
4 2
2 1
3 2
4 2
3 4
4 2
Sample Output
2
2
HINT
请使用printf语句进行输出.用cout会TLE
解析
这个题如果一个一个的往上找的话会超时,所以我们就翻倍的往上找。还有一个就是我们只能找不同的那一个,因为我们是翻倍找到如果直接找相同的可能不是最近的那一个。
程序如下
#include<bits/stdc++.h>
#define N 1000002
int n,q,rt,f[N][20],dep[N],a[N],pre[N],now[N],son[N],tot;//f[i][j]表示从i点出发走2^j到的点
bool bl[N];
using namespace std;
void put(int x,int y)// 存入这条边
{
pre[++tot]=now[x];
now[x]=tot;
son[tot]=y;
}
void dfs(int u,int d)//这个是做树的深度
{
dep[u]=d;//u的深度为d
bl[u]=1;//打个标记,走过了
for(int i=now[u];i;i=pre[i])//把点枚举一遍
{
int c=son[i];
if(!bl[c])//如果没走过
{
f[c][0]=u;//他的父亲点为u,也就是走2^0,就是1
dfs(c,d+1);
}
}
}
int lca(int x,int y)
{
if(dep[x]<dep[y])//如果x的深度小于y的,就换一下
swap(x,y);
for(int i=19;i>=0;i--)
if(dep[f[x][i]]>=dep[y])
x=f[x][i];
if(x==y)return x;//如果跳到了同一个点,就直接return x
for(int i=19;i>=0;i--)//逆序枚举
if(f[x][i]!=f[y][i])//如果是跳到两个不同的点,注意是不同的,如果相同可能不是最近的
x=f[x][i],y=f[y][i];
return f[x][0];//父亲点就是答案
}
int main()
{
int x,y;
scanf("%d%d",&n,&q);
rt=1;
for(int i=1,x,y;i<n;i++)
{
scanf("%d%d",&x,&y);
put(x,y);
put(y,x);
}
dfs(rt,1);
for(int i=1;i<=n;i++)
for(int j=1;j<20;j++)
f[i][j]=f[f[i][j-1]][j-1];//整理出所有的向上跳2^j步到哪个点
for(int i=1;i<=q;i++)
{
scanf("%d%d",&x,&y);
printf("%d\n",lca(x,y));
}
return 0;
}