codevs——1503 愚蠢的宠物

1503 愚蠢的宠物

时间限制: 1 s

空间限制: 128000 KB

题目等级 : 黄金 Gold

查看运行结果

题目描述 Description

大家都知道，sheep有两只可爱的宠物（一只叫神牛，一只叫神菜）。有一天，sheep带着两只宠物到狗狗家时，这两只可爱的宠物竟然迷路了……

狗狗的家因为常常遭到猫猫的攻击，所以不得不把家里前院的路修得非常复杂。狗狗家前院有N个连通的分叉结点，且只有N-1条路连接这N个节点，节点的编号是1-N（1为根节点）。sheep的宠物非常笨，他们只会向前走，不会退后（只向双亲节点走），sheep想知道他们最早什么时候会相遇（即步数最少）。

输入描述 Input Description

第1行：一个正整数N，表示节点个数。

第2~N行：两个非负整数A和B，表示A是B的双亲。（保证A,B<=n）

第N+1行：两个非负整数A和B，表示两只宠物所在节点的位置。（保证A,B<=n）

输出描述 Output Description

输出他们最早相遇的节点号。

样例输入 Sample Input

10
1 2
1 3
1 4
2 5
2 6
3 7
4 8
4 9
4 10
3 6

样例输出 Sample Output

数据范围及提示 Data Size & Hint

对于10%的数据，n<10^6

对于100%的数据，n<=10^6

lca模板

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define N 1100000
using namespace std;
bool vis[N];
int n,root,x,y,tot;
int dad[N],head[N],deep[N],size[N],fa[N],top[N];
struct Edge
{
    int next,to,from;
}edge[N];
int add(int x,int y)
{
    tot++;
    edge[tot].to=y;
    edge[tot].next=head[x];
    head[x]=tot;
}
int read()
{
    ,f=; char ch=getchar();
    ; ch=getchar();}
    +ch-'; ch=getchar();}
    return x*f;
}
int dfs(int x)
{
    size[x]=;vis[x]=true;
    deep[x]=deep[fa[x]]+;
    for(int i=head[x];i;i=edge[i].next)
    {
        int to=edge[i].to;
        if(fa[x]!=to&&!vis[to])
        {
            fa[to]=x;
            dfs(to);
            size[x]+=size[to];
        }
    }
}
int dfs1(int x)
{
    ;vis[x]=true;
    if(!top[x]) top[x]=x;
    for(int i=head[x];i;i=edge[i].next)
    {
        int to=edge[i].to;
        if(fa[x]!=to&&size[t]<size[to])
           t=to;
    }
    if(t&&!vis[t])
    {
        top[t]=top[x];
        dfs1(t);
    }
    for(int i=head[x];i;i=edge[i].next)
      if(fa[x]!=edge[i].to&&edge[i].to!=t&&!vis[edge[i].to])
       dfs1(edge[i].to);
}
int lca(int x,int y)
{
    while(top[x]!=top[y])
    {
        if(deep[top[x]]<deep[top[y]])
         swap(x,y);
        x=fa[x];
    }
    if(deep[x]>deep[y])
     swap(x,y);
    return x;
}
int main()
{
    n=read();
    ;i<n;i++)
    {
        x=read(),y=read();
        add(x,y),add(y,x);
        dad[y]=x;
    }
    ;i<=n;i++)
     if(!dad[i]) root=i;
    dfs(root);
    memset(vis,,sizeof(vis));
    dfs1(root);
    x=read(),y=read();
    printf("%d",lca(x,y));
}

巴特西

codevs——1503 愚蠢的宠物

1503 愚蠢的宠物

最新文章

热门文章