AcWing 3555. 二叉树#LCA朴素算法

AcWing 3555. 二叉树#LCA朴素算法_第1张图片


 本蒟蒻以为深度可以直接得出....即d[儿子节点]=d[父亲节点]+1,但是由于输入是无序的,即给定的树不是从根节点开始往下扩展,而是随便给的,比如

 

AcWing 3555. 二叉树#LCA朴素算法_第2张图片

这样根据式子得出的答案必定是错误的,只能等树全部建完之后在进行dfs,确定深度。

AC代码

#include 
using namespace std;

int T,n,m,a,b,d[1001],f[1001];
pairnode[1001];
void dfs(int now,int depth){
    d[now]=depth;
    if(node[now].first!=-1)dfs(node[now].first,depth+1);
    if(node[now].second!=-1)dfs(node[now].second,depth+1);
}
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;++i){
            scanf("%d%d",&a,&b);
            if(a!=-1)f[a]=i;//d[a]=d[i]+1 是不可行的
            if(b!=-1)f[b]=i;
            node[i]={a,b};
        }
        dfs(1,1);
        for(int i=1,res;i<=m;++i){
            res=0;
            scanf("%d%d",&a,&b);
            while(d[a]>d[b])a=f[a],res++;
            while(d[b]>d[a])b=f[b],res++;
            while(a!=b)a=f[a],b=f[b],res+=2;
            printf("%d\n",res);
        }
    }
    return 0;
}

你可能感兴趣的:(AcWing,算法,数据结构,c++,LCA)