1431: LCA 最近公共祖先
时间限制: 1 Sec
内存限制: 128 MB
提交: 244
解决: 36
[
提交][
状态]
1431: LCA 最近公共祖先
时间限制: 1 Sec 内存限制: 128 MB提交: 244 解决: 36
[ 提交][ 状态]
题目描述
给一棵树, 节点数为N(1 <= N <= 250,000), 和Q(0 <= Q <= 100,000)个询问, 对于每个询问求出所求两点的最近公共祖先
输入
第一行: 节点数N
以下N行, 第i + 1行: 点i的父亲节点Father[i] (假定根的父亲是0)
下一行为询问数Q
每个询问包含两个整数i, j你的程序应按照询问的次序回答出所求两点的最近公共祖先
输出
共N行, 第i行: 第i个询问的答案
样例输入
12
0
1
1
1
2
2
3
3
8
8
9
10
5
5 12
12 7
9 12
10 12
12 12
样例输出
1
3
8
10
12
倍增算法:主要分两步。
第一步:将两个节点"窜"到一个深度
第二步:两个节点一起"窜"到他们的lca(最近公共祖先)。
'窜'的操作,如果你每个节点一下,一下,一下的窜,会非常慢。。所以就有了倍增,要一次窜2^n个节点。当然如果过头了就少窜一点。如果你写过01
背包,它的思路和2进制拆包很像。
看起来好像很简单的样子>-<代码如下(带注释)
#include<iostream>
#include<cmath>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<ctime>
using namespace std;
const int maxn=250020;
int n,q;
int g[100][100],ne,root;
int p[maxn][20];//p[i][j]表示i结点的第2^j祖先
int deep[maxn];//i的深度
int use[1000][1000];
void build(int x)//处理树深度
{
for(int i=1;i<=n;i++)
if(g[x][i]==1)
{
deep[i]=deep[x]+1;
p[i][0]=x;
g[x][i]=g[i][x]=0;
build(i);
}
}
void pre()//处理祖先
{
int i,j;
for(j=1;(1<<j)<=n;j++)
for(i=1;i<=n;i++)//! !j在i前面!!记住
if(p[i][j-1]!=-1)
p[i][j]=p[p[i][j-1]][j-1];//i的第2^j祖先就是i的第2^(j-1)祖先的第2^(j-1)祖先
}
int lca(int a,int b)//最近公共祖先
{
//使a,b两点的深度相同
int i,j;
if(deep[a]<deep[b])swap(a,b);
for(i=0;(1<<i)<=deep[a];i++);
i--;
for(j=i;j>=0;j--)
if(deep[a]-(1<<j)>=deep[b])
a=p[a][j];
if(a==b)return a;
//倍增法,每次向上进深度2^j,找到最近公共祖先的子结点
for(j=i;j>=0;j--)
{
if(p[a][j]!=-1&&p[a][j]!=p[b][j])//!!
{
a=p[a][j];
b=p[b][j];
}
}
return p[a][0];//即正好窜到两个子树上,最近祖先为它的j=0的祖先
}
int main()
{
cin>>n>>q;
for(int i=1;i<=n;i++)
{
int a,b;
cin>>a>>b;
g[a][b]=g[b][a]=1;
}
p[1][0]=-1;
deep[1]=1;
build(1);
pre();
for(int i=1;i<=q;i++)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%d\n",lca(x,y));
}
return 0;
}