lca最近公共祖先(st表/倍增)

大体思路

1.求出每个元素在树中的深度

2.用st表预处理的方法处理出f[i][j],f[i][j]表示元素i上方第2^j行对应的祖先是谁 

3.将较深的点向上挪,直到两结点的深度相同

4.深度相同后,祖先可能就在上方,再走几步就到了,于是两个点同时向上移

具体的方法和代码贴在下面 ↓

具体

1.求出每个元素在树中的深度

//求每个节点在树中的深度
void dfs(int pos,int pre)//pre是pos的父节点 
{
    for(int i=0;i<v[pos].size;i++)//枚举pos的子节点 
    {
        register int t=v[pos][i];
        if(t==pre)continue;//防止死循环 
        f[t][0]=pos;dep[t]=dep[pos]+1;
        dfs(t,pos);//再从子节点向后枚举 
    }
}

2.用st表预处理的方法处理出f[i][j]

//求f数组(st表预处理) 
for(int i=1;(1<<i)<=n;i++)
    for(int j=1;j<=n;j++)
        f[j][i]=f[f[j][i-1]][i-1];
//f[i][j]表示元素i上方第2^j行对应的祖先是谁 

3.先比较a,b两点哪个较深,将较深的点赋值给a

//把a节点变为a,b中较深的一个节点 
int query(int a,int b)
{
    if(dep[a]<dep[b])swap(a,b);
}

将较深的点向上挪,直到两结点的深度相同

//使a和b在同一个深度上 
for(int i=20;i>=0;i--)
    if(dep[f[a][i]]>=dep[b])
        a=f[a][i];
//倒着循环是因为向上走的步数只会越来越小 

4.深度相同后,祖先可能就在上方,再走几步就到了,于是两个点同时向上移

//同一深度后,再向上找公共祖先 
for(int i=20;i>=0;i--)
    if(f[a][i]!=f[b][i])
    {
        a=f[a][i];
        b=f[b][i];
     } 

 全部代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
vector<int> v[41000];
vector<int> w[41000];
int f[41000][21];//f[i][j]表示i点向上2^j层的祖先 
int g[41000][21];//g[i][j]表示i点到从i向上2^j层的祖先的距离 
int dep[41000];
int n,m;
void dfs(int pos,int pre,int depth)
{
    dep[pos]=depth;
    for(int i=0;i<v[pos].size();i++)
    {
        int t=v[pos][i];
        if(t==pre) continue;
        f[t][0]=pos;
        g[t][0]=w[pos][i];
        dfs(t,pos,depth+1);
    }
}
int query(int a,int b)
{
    int sum=0;
    if(dep[a]<dep[b]) swap(a,b);//深度较深的点 
    for(int i=20;i>=0;i--)//找到a在深度dep[b]处的祖先 
    {
        if(dep[f[a][i]]>=dep[b])
        {
            sum+=g[a][i];//a到该祖先的距离 
            a=f[a][i];
        }
    }
    if(a==b) return sum;//挪到相同深度后如果在同一点直接return 
    int x;
    for(int i=20;i>=0;i--)//否则a和b一起往上蹦跶 
    {
        if(f[a][i]!=f[b][i])
        {
            sum+=g[a][i];
            a=f[a][i];
            sum+=g[b][i];
            b=f[b][i];
        }
    }
    return sum+g[a][0]+g[b][0];//最后蹦跶到最近公共祖先的下一层,所以要再加上上一层 
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        scanf("%d%d",&n,&m);
        memset(dep,-1,sizeof dep);//多组数据我们初始化 
        memset(f,0,sizeof f);
        memset(g,0,sizeof g);
        for(int i=0;i<n;i++)//md
            v[i].clear(),w[i].clear();
        for(int i=1;i<n;i++)
        {
            int x,y,c;
            cin>>x>>y>>c;
            v[x].push_back(y);
            w[x].push_back(c);
            v[y].push_back(x);
            w[y].push_back(c);
        }
        int xxx=v[1].size();
        dfs(1,0,1);//dfs处理出每个点的深度,以及各种... 
    
        for(int i=1;1<<i<=n;i++)
            for(int j=1;j<=n;j++)
                f[j][i]=f[f[j][i-1]][i-1],
                g[j][i]=g[f[j][i-1]][i-1]+g[j][i-1];
        for(int i=1;i<=m;i++)
        {
            int x,y;
            cin>>x>>y;
            if(x==y) cout<<"0"<<endl;
            else cout<<query(x,y)<<endl;
        }
    }
    return 0;
}

 

posted @ 2017-04-14 09:37  Echo宝贝儿  阅读(234)  评论(0编辑  收藏  举报