0921 LCA练习
阅读原文时间:2023年07月08日阅读:3

数据结构中的树,在计算机科学中是非常重要的,例如我们来看看下面这棵树:

在图中我们对每个节点都有编号了。 8号节点是这棵树的根。我们定义,一个子节点向它的根节点的路径上,任意一个节点都称为它的祖先。例如, 4号节点是16号节点的祖先。而10号节点同样也是16号的祖先。事实上,16号的祖先有8,4,10,16共四个。另外8, 4, 6,7都是7号节点的祖先,所以7号和16号的公共祖先是4和8号,而在这两个里面,4号是距离7和16最近的一个,所以我们称7号和16号的最近公共祖先是4号。

再例如,2和3的最近公共祖先是10,再例如6和13的是8。

现在你需要编写一个程序,在一棵树中找出指定两个节点的最近公共祖先

Input

第一行输入T表示有T组数据。每组第一行是N表示这棵树有多少个节点,其中 2<=N<=10,000。 节点用正整数1, 2,..., N表示。 接下来的 N -1 行表示这棵树的边,每行两个数,都是节点编号,前一个是后一个的父节点。最后一行是要查询的两个节点,计算出这两个节点的最近公共祖先

Output

对于每组测试输出一行,输出它们的最近公共祖先的编号。

分析:lca

代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;
#define maxn 10005
int n,m,s;
int cnt,head[maxn],depth[maxn],p[maxn][25],fat[500005];
struct node{
    int nxt,to;
}e[maxn];
void add(int x,int y){
    e[++cnt].nxt=head[x];
    head[x]=cnt;
    e[cnt].to=y;
    fat[y]=x;
}
void pre(int u,int fa){
    depth[u]=depth[fa]+1;
    p[u][0]=fa;
    for(int i=1;1<<i<=depth[u];i++)p[u][i]=p[p[u][i-1]][i-1];
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v!=fa)pre(v,u);
    }
}
int lca(int a,int b){
    if(depth[a]>depth[b])swap(a,b);
    for(int i=20;i>=0;i--){
        if(depth[a]<=depth[b]-(1<<i))b=p[b][i];
    }
    if(a==b)return a;
    for(int i=20;i>=0;i--){
        if(p[a][i]==p[b][i])continue;
        else a=p[a][i],b=p[b][i];
    }
    return p[a][0];
}
int main(){
    int t;scanf("%d",&t);
    while(t--){
        memset(head,0,sizeof(head));
        memset(depth,0,sizeof(depth));
        memset(p,0,sizeof(p));
        memset(fat,0,sizeof(fat));
        memset(e,0,sizeof(e));
        cnt=0;
        scanf("%d",&n);
        for(int i=1;i<n;i++){
            int x,y;scanf("%d%d",&x,&y);
            add(x,y);
        }
        for(int i=1;i<=n;i++){
            if(!fat[i]){
                s=i;
                break;
            }
        }
        pre(s,0);
        int a,b;scanf("%d%d",&a,&b);
        printf("%d\n",lca(a,b));
    }
    return 0;
}

题目描述

欢乐岛上有个非常好玩的游戏,叫做“紧急集合”。在岛上分散有 nnn 个等待点,有 n−1n-1n−1 条道路连接着它们,每一条道路都连接某两个等待点,且通过这些道路可以走遍所有的等待点,通过道路从一个点到另一个点要花费一个游戏币。

参加游戏的人三人一组,开始的时候,所有人员均任意分散在各个等待点上(每个点同时允许多个人等待),每个人均带有足够多的游戏币(用于支付使用道路的花费)、地图(标明等待点之间道路连接的情况)以及对话机(用于和同组的成员联系)。当集合号吹响后,每组成员之间迅速联系,了解到自己组所有成员所在的等待点后,迅速在 nnn 个等待点中确定一个集结点,组内所有成员将在该集合点集合,集合所用花费最少的组将是游戏的赢家。

小可可和他的朋友邀请你一起参加这个游戏,由你来选择集合点,聪明的你能够完成这个任务,帮助小可可赢得游戏吗?

输入格式

第一行两个正整数 nnn 和 mmm,分别表示等待点的个数(等待点也从 111 到 nnn 进行编号)和获奖所需要完成集合的次数。随后 n−1n-1n−1 行,每行两个正整数 a,ba,ba,b,表示编号为 aaa 和编号为 bbb 的等待点之间有一条路。随后 mmm 行,每行用三个正整数 x,y,zx,y,zx,y,z,表示某次集合前小可可、小可可的朋友以及你所在等待点的编号。

输出格式

输出共 mmm 行,每行两个用空格隔开的整数 p,cp,cp,c。其中第 iii 行表示第 iii 次集合点选择在编号为 ppp 的等待点,集合总共的花费是 ccc 个游戏币。

分析:

经分析,对每两个点求lca,三个点一定有其中两个lca相同,并在另一个lca上面,下面的lca一定是到三点路径并最小的,正确性显然。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define maxn 500005
int n,m,s;
int cnt,head[maxn],depth[maxn],p[maxn][25],fat[maxn];
struct node{
    int nxt,to;
}e[2*maxn];
void add(int x,int y){
    e[++cnt].nxt=head[x];
    head[x]=cnt;
    e[cnt].to=y;
}
void pre(int u,int fa){
    depth[u]=depth[fa]+1;
    fat[u]=fa;
    p[u][0]=fa;
    for(int i=1;1<<i<=depth[u];i++)p[u][i]=p[p[u][i-1]][i-1];
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v!=fa)pre(v,u);
    }
}
int lca(int a,int b){
    if(depth[a]>depth[b])swap(a,b);
    for(int i=20;i>=0;i--){
        if(depth[a]<=depth[p[b][i]])b=p[b][i];//printf("$$$$%d %d\n",i,b);
    }
    //printf("%d %d\n",a,b);
    if(a==b)return a;
    for(int i=20;i>=0;i--){
        if(p[a][i]==p[b][i])continue;
        else a=p[a][i],b=p[b][i];
    }
    return p[a][0];
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        int a,b;scanf("%d%d",&a,&b);
        add(a,b);add(b,a);
    }
    pre(1,0);
    //printf("sidjhsdl%d\n",depth[0]);
    //for(int i=1;i<=n;i++){
    // printf("%d",depth[i]);
    // }
    for(int i=1;i<=m;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        int aa=lca(x,y);
        int bb=lca(x,z);
        int cc=lca(y,z);
        // printf("%d %d %d\n",aa,bb,cc);
        int t;
        if(aa==bb)t=cc;
        if(aa==cc)t=bb;
        if(bb==cc)t=aa;
        printf("%d %d\n",t,depth[x]+depth[y]+depth[z]-depth[aa]-depth[bb]-depth[cc]);
    }
    return 0;
}