PID28 [Stupid]愚蠢的宠物

题链:https://www.rqnoj.cn/problem/28

题目描述

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

描述

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

输入格式

第1行:一个正整数N,表示节点个数。
第2~N行:两个非负整数A和B,表示A是B的双亲。(保证A,B<=n)
第N+1行:两个非负整数A和B,表示两只宠物所在节点的位置。(保证A,B<=n)

输出格式

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

因为只有一次询问,所以算出这两个节点到他们祖先节点的距离,然后枚举节点作为公共祖先即可

#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <vector>
#define ll long long
#define inf 1000000000LL
#define mod 1000000007
using namespace std;
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;
}
const int N=1000005;
int fa[N],dis1[N],dis2[N];
int main()
{
    int n=read();
    for(int i=1;i<n;i++){
        int a=read(),b=read();
        fa[b]=a;
    }
    int a=read(),b=read();
    int f1,f2;
    memset(dis1,0x3f,sizeof(dis1));
    memset(dis2,0x3f,sizeof(dis2));
    dis1[a]=dis2[b]=0;
    f1=a;f2=b;
    for(int i=1;i<=n;i++){
        f1=fa[f1];f2=fa[f2];
        dis1[f1]=min(dis1[f1],i);
        dis2[f2]=min(dis2[f2],i);    
    }
    int ans=inf,id=1;
    for(int i=1;i<=n;i++){
        if(ans>dis1[i]+dis2[i]){
            ans=dis1[i]+dis2[i];
            id=i;
        }
    }
    printf("%d
",id);
    return 0;
}

原文地址:https://www.cnblogs.com/zsyacm666666/p/6954100.html