当前位置: 首页 > 工具软件 > POG > 使用案例 >

HDU 5266 pog loves szh III [lca 倍增算法]

尚安平
2023-12-01

之前我们在面对一个查询的时候,直接采用最暴力的搜索去完成此工作,现在,当我们面对有很多组数据的时候,发现一个一个的查询效率实在是太慢了,所以我们采用了一种新的方式,那就是倍增

这个题就是给定Q个查询,查询一棵树上两个点的LCA

Q<3e5 and N<3e5

这时候,我们就可以采取倍增的做法,我们原本是一个一个向上找祖父结点,但是这样太慢了,如果一直我们向上的距离是d的话,那么我们就可以把D 分解成 2^0 ,2^1,2^2,…,2^n 这些数的组合,那么这样我们就可以从大到小向上跳跃,跳到某个祖父节点,判断 是否能够满足条件,就可以了。这个题还要注意的是双向边,然后dfs的时候从1开始就可以了

#include <bits/stdc++.h>
#define cl(a) memset(a,0,sizeof(a))
using namespace std;
const int maxn=3e5+500;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
typedef long long ll;
int flag[maxn];//深度遍历时用于标记节点是否被访问
int _father[maxn];//标记每个节点的父亲节点
int depth[maxn];
vector<int>g[maxn];
int p[maxn][25];
int T,n,m;
void init()
{
    memset(p,-1,sizeof(p));
    cl(flag);
    cl(depth);
    cl(_father);
    for(int i=0; i<maxn; i++)
    {
        g[i].clear();
    }

}
void build(int x)
{
    flag[x]=1;
    for(int i=0;i<g[x].size();i++)
    {
        int v=g[x][i];
        if(!flag[v])
        {
        depth[v]=depth[x]+1;
        p[v][0]=x;
        build(v);
        }
    }
}
void pre()
{
    int i,j;
    for(j=1; (1<<j)<=n; j++)
    {
        for(i=1; i<=n; i++)
        {
            if(p[i][j-1]!=-1)
            {
                p[i][j]=p[p[i][j-1]][j-1];
            }
        }
    }
}
int lca(int a,int b)
{
    int i,j;
    if(depth[a]<depth[b])swap(a,b);
    for(i=0; (1<<i)<=depth[a]; i++);
    i--;
    for(j=i; j>=0; j--)
    {
        if(depth[a]-(1<<j)>=depth[b])
            a=p[a][j];
    }
    if(a==b)return a;
    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];
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    while(cin>>n)
    {
        init();

        for(int i=1; i<n; i++)
        {
            int x,y;
            cin>>x>>y;
            g[x].push_back(y);
            g[y].push_back(x);

        }

        build(1);
        pre();

        cin>>m;
        for(int i=1; i<=m; i++)
        {
            int u,v;
            cin>>u>>v;
            cout<<lca(u,v)<<endl;
        }

    }
    return 0;
}
 类似资料: