【倍增LCA】TZOJ 5701 最近公共祖先

题面：给定一棵树，请查询结点u和v的最近公共祖先。最近公共祖先，就是两个节点在这棵树上深

度最大（离根结点最远）的公共的祖先节点，结点的祖先也可以是自身。

输入：

输入数据第一行为结点个数n（n<=900），接下来有n行，每行格式如下：

x m y1, y2, ... ym

表示结点x有m个孩子y1, y2, ... ym

接下来有一个正整数q(q<=100000)，表示查询次数，之后有q行，每行两个正整数u和v，表示待查寻的结点编号。

树结点的编号为1~n。

输出：

对每次查询，输出u和v的最近公共祖先，格式为：

u v = w

其中w表示u和v的最近公共祖先结点编号。

代码：

 

#include<bits/stdc++.h>
using namespace std;

int n,m,t,d;
vector<int>vec[905]; //堆记录子节点，也可以是链式向前星(较麻烦)
int dep[905],dp[905][25]; //dep记录点深度，dp[i][j] i为当前点，j为层数，dp代表了i向上j+1层的父节点(ST算法)

void dfs(int root,int pre)
{
    dep[root]=dep[pre]+1; //记录深度
    dp[root][0]=pre; //记录父节点
    for(int i=1;(1<<i)<=dep[root];i++)
        dp[root][i]=dp[dp[root][i-1]][i-1]; //该点往上i层等于该点往上1层再往上i-1层，借助上层已确定的点找点
    for(int i=0;i<vec[root].size();i++) //题目给出的可以确保是层层向下的
        if(vec[root][i]!=pre)
            dfs(vec[root][i],root);
}

int lca(int a,int b)
{
    if(dep[b]>dep[a]) //确保a是较大的一个
        swap(a,b);
    for(int i=23;i>=0;i--) //让a转移到与b最接近的位置
        if(dep[a]-(1<<i)>=dep[b]) //步步逼近
            a=dp[a][i];
    if(a==b) return a;
    for(int i=23;i>=0;i--)
    {
        if(dp[a][i]!=dp[b][i])
        {
            a=dp[a][i];
            b=dp[b][i];
        }
    }
    return dp[a][0]; //返回a的父节点
}

int main()
{
    cin>>t;
    for(int i=1;i<=t;i++)
    {
        cin>>n>>m;
        while(m--)
        {
            cin>>d;
            vec[n].push_back(d);
        }
    }
    for(int i=1;i<=t;i++) //遍历找到最高的节点
        if(!dep[i])
            dfs(i,0);
    cin>>t;
    while(t--)
    {
        int a,b;
        cin>>a>>b;
        printf("%d %d = ",a,b);
        cout<<lca(a,b)<<endl;
    }
    // system("pause");
}