LCA-倍增法(在線)O(nlogn)-O(logn)


1. DFS預處理出所有節點的深度和父節點
inline void dfs(int u)
{
    int i;
    for(i=head[u];i!=-1;i=next[i])  
    {  
        if (!deep[to[i]])
        {            
            deep[to[i]] = deep[u]+1;
            p[to[i]][0] = u; //p[x][0]保存x的父節點為u;
            dfs(to[i]);
        }
    }
}

 

2. 初始各個點的2^j祖先是誰 ,其中2^j(j=0...log(該點深度))倍祖先,1倍祖先就是父親,2倍祖先是父親的父親......。

void init()
{
    int i,j;
    //p[i][j]表示i結點的第2^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];//i的第2^j祖先就是i的第2^(j-1)祖先的第2^(j-1)祖先
}

 

3.從深度大的節點上升至深度小的節點同層,如果此時兩節點相同直接返回此節點,即lca。
否則,利用倍增法找到最小深度的 p[a][j]!=p[b][j],此時他們的父親p[a][0]即lca。
int lca(int a,int b)//最近公共祖先
{
    int i,j;
    if(deep[a]<deep[b])swap(a,b);
    for(i=0;(1<<i)<=deep[a];i++);
    i--;
    //使a,b兩點的深度相同
    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];
}

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM