1787: [Ahoi2008]Meet 紧急集合

最新推荐文章于 2018-01-14 20:49:00 发布

原创最新推荐文章于 2018-01-14 20:49:00 发布 · 169 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

----LCA

本文提供了一道名为[Ahoi2008]Meet紧急集合的算法题目解答，采用C++实现，主要介绍了如何通过深度优先搜索（DFS）建立树状结构并求解三个节点间的最近公共祖先（LCA），最终输出节点间距离。

1787: [Ahoi2008]Meet 紧急集合

Time Limit: 20 Sec Memory Limit: 162 MB
Submit: 3217 Solved: 1445
[ Submit][ Status][ Discuss]

Description

Input

Output

Sample Input

6 4
1 2
2 3
2 4
4 5
5 6
4 5 6
6 3 1
2 4 4
6 6 6

Sample Output

5 2
2 5
4 1
6 0

HINT

题解：同样模板啊，没什么好讲的，直接上代码。

代码：

#include<bits/stdc++.h>
using namespace std;
int n,q,cnt;
int deep[500001],head[500001],fa[500001][20];
bool vis[500001];
struct node
{
    int to,next;
}edge[1000001];
int read()
{
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void ins(int u,int v)
{
    edge[++cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt;
}
void dfs(int x)
{
    vis[x]=1;
    for(int i=1;i<=18;i++)
    {
        if(deep[x]<(1<<i))break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }
    for(int i=head[x];i;i=edge[i].next)
    {
        if(vis[edge[i].to])continue;
        deep[edge[i].to]=deep[x]+1;
        fa[edge[i].to][0]=x;
        dfs(edge[i].to);
    }
}
int lca(int x,int y)
{
    if(deep[x]<deep[y])swap(x,y);
    int d=deep[x]-deep[y];
    for(int i=0;i<=18;i++)
        if((1<<i)&d)x=fa[x][i];
    for(int i=18;i>=0;i--)
        if(fa[x][i]!=fa[y][i]) 
            x=fa[x][i],y=fa[y][i];
    if(x==y)return x;
    else return fa[x][0];
}
int dis(int x,int y)
{
    return deep[x]+deep[y]-2*deep[lca(x,y)];
}
int cal(int x,int y,int z)
{
    int p1=lca(x,y),p2=lca(x,z),p3=lca(y,z),t;
    if(p1==p2)t=p3;
    else
        if(p2==p3)t=p1;
        else t=p2;
    int ans;
    ans=dis(x,t)+dis(y,t)+dis(z,t);
    printf("%d %d\n",t,ans);
}
int main()
{
    n=read();q=read();
    for(int u,v,i=1;i<n;i++)
    {
        u=read();v=read();
        ins(u,v),ins(v,u);
    }
    dfs(1);
    for(int i=1,x,y,z;i<=q;i++)
        x=read(),y=read(),z=read(),cal(x,y,z);
    return 0;
}

标签

#LCA 求最近公共祖先