标签：剖分 int son LCA 树链 define

最近刚好学了树链剖分，来一道LCA的模板题练练手：P3379 最近公共祖先（LCA）

题意：给一棵多叉树，m次询问，求两个节点之间的LCA。

思路：树链剖分。

#include <bits/stdc++.h>
#include<cstring>
#define L(x) (x<<1)
#define R(x) (x<<1|1)
#define fori(a,b,c) for(int a=b;a<=c;a++)
#define ford(a,b,c) for(int a=c;a>=b;a--)
#define mem0(x) memset(x,0,sizeof(x))
#define mem1(x) memset(x,-1,sizeof(x))
using namespace std;
const int M=5e5+100;


vector<int> g[M];
int son[M],top[M],f[M],sz[M],dep[M],tot;

void dfs1(int s,int fa)
{
    f[s]=fa;
    dep[s]=dep[fa]+1;
    sz[s]=1;
    int ma=-1;
    for(int i=0;i<g[s].size();i++)
    {
        int v=g[s][i];
        if(v==fa)continue;
        dfs1(v,s);
        sz[s]+=sz[v];
        if(sz[v]>ma)
        {
            ma=sz[v];
            son[s]=v;
        }
    }
}

void dfs2(int s,int t)
{
    top[s]=t;
    if(!son[s])return;
    dfs2(son[s],t);
    for(int i=0;i<g[s].size();i++)
    {
        int v=g[s][i];
        if(v==f[s]||v==son[s])continue;
        dfs2(v,v);
    }
}

int fd(int a,int b)
{
    while(top[a]!=top[b])
    {
        if(dep[top[a]]<dep[top[b]])swap(a,b);
        a=f[top[a]];
    }
    if(dep[a]<dep[b])swap(a,b);
    return b;
}

int main()
{
    int n,m,r,x,y;
    scanf("%d%d%d",&n,&m,&r);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs1(r,r);
    dfs2(r,r);
    while(m--)
    {
        scanf("%d%d",&x,&y);
        printf("%d\n",fd(x,y));
    }
    return 0;
}

标签：剖分,int,son,LCA,树链,define
来源： https://blog.csdn.net/qq_41365703/article/details/99684582

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。