欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

P3398 仓鼠找sugar ( 倍增+思维 )

程序员文章站 2022-06-16 11:17:16
...

题目链接

P3398 仓鼠找sugar ( 倍增+思维 )
解题报告:
题目给出树形结构,走最短路径,很显然跟最近公共祖先有关。
树形结构中若有两条路径相交,那么一定满足一条链的LCA在另一条链上,充分必要条件
P3398 仓鼠找sugar ( 倍增+思维 )
路径相交图如上,图一显然不服符合树形结构,每个节点只有一个父节点,
图二图三就是合法相交的情况;满足图二图三之一均成立。
根据树上路径唯一性:
图二:dist(c,LCA2)+dist(LCA2,d)=dist(c,d)
图三:dist(a,LCA1)+dist(LCA1,b)=dist(a,b)

#define first f
#define second s
#define ll long long
#define mp make_pair
#define pb push_back
#define pf push_front
#define lb lower_bound
#define ub upper_bound
#include <bits/stdc++.h>
#define pii pair<string,string>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int MOD=1e9+7;
const int inf=1e9+7;
const int maxn=1e5+5;
const double eps=1e-6;
const double PI=acos(-1.0);
const double e=2.718281828459;

struct node
{
    int to,next;
}edge[maxn<<1];
int head[maxn],cnt;
int deep[maxn],f[maxn][25];
void add(int from,int to)
{
    edge[++cnt].next=head[from];
    edge[cnt].to=to;
    head[from]=cnt;
}
void dfs(int now,int pre)
{
    deep[now]=deep[pre]+1;
    f[now][0]=pre;
    for(int i=1;i<=20;i++){
        f[now][i]=f[f[now][i-1]][i-1];
    }
    for(int i=head[now];~i;i=edge[i].next){
        int to=edge[i].to;
        if(to==pre){continue;}
        dfs(to,now);
    }
}
int LCA(int x,int y)
{
    if(deep[x]<deep[y]){swap(x,y);}
    for(int i=20;i>=0;i--){
        if(deep[f[x][i]]>=deep[y]){
            x=f[x][i];
        }
    }
    if(x==y){return x;}
    for(int i=20;i>=0;i--){
        if(f[x][i]!=f[y][i]){
            x=f[x][i];
            y=f[y][i];
        }
    }
    return f[x][0];
}
int dist(int x,int y)
{
    return deep[x]+deep[y]-2*deep[LCA(x,y)];
}
int main()
{
    mem(head,-1);
    int n,m,u,v;
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    }
    dfs(1,0);
    int a,b,c,d;
    while(m--){
        scanf("%d%d%d%d",&a,&b,&c,&d);
        int id1=LCA(a,b);
        int id2=LCA(c,d);
        if(dist(c,id1)+dist(id1,d)==dist(c,d)||dist(a,id2)+dist(id2,b)==dist(a,b)){
            printf("Y\n");
        }
        else{printf("N\n");}
    }
    return 0;
}