【lca】P3398 仓鼠找 sugar

逃离我推掉我的手 2024-03-27 09:03 44阅读 0赞

[P3398 仓鼠找 sugar - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)][P3398 _ sugar - _ _ _ _luogu.com.cn]

题意：

![63205ab51fc33ab641ff3c43f9c8f4ed.png][]

思路：

树上两条路径相交的充要条件是：其中一条链的lca在另一条链的路径上

![4cf0e5e55562545470e77604d20b476f.png][]

Code：

#include <bits/stdc++.h>
    using namespace std;
    const int mxe=1e5+10,mxn=1e6+10;
    struct ty{
        int to,next;
    }edge[mxe<<1];
    queue<int> q;
    int n,Q,u,v,tot=0,a,b,c,d;
    int head[mxn],dep[mxn],vis[mxn],f[mxn][65];
    void add(int u,int v){
        edge[tot].to=v;
        edge[tot].next=head[u];
        head[u]=tot++;
    }
    void init(){
        tot=0;
        for(int i=0;i<=n;i++){
            head[i]=-1;
        }
    }
    void bfs(){
        memset(vis,0,sizeof(vis));
        memset(dep,0x3f,sizeof(dep));
        dep[0]=0;dep[1]=1;
        q.push(1);
        vis[1]=1;
        while(!q.empty()){
            int u=q.front();
            q.pop();
            for(int i=head[u];~i;i=edge[i].next){
                if(vis[edge[i].to]) continue;
                vis[edge[i].to]=1;
                dep[edge[i].to]=dep[u]+1;
                f[edge[i].to][0]=u;
                for(int j=1;j<=30;j++){
                    f[edge[i].to][j]=f[f[edge[i].to][j-1]][j-1];
                }
                q.push(edge[i].to);
            }
        }
    }
    int lca(int u,int v){
        if(dep[u]<dep[v]) swap(u,v);
        for(int j=30;j>=0;j--){
            if(dep[f[u][j]]>=dep[v]){
                u=f[u][j];
            }
        }
        if(u==v) return u;
        for(int j=30;j>=0;j--){
            if(f[u][j]!=f[v][j]){
                u=f[u][j];
                v=f[v][j];
            }
        }
        return f[u][0];
    }
    int dis(int x,int y){
        int t=lca(x,y);
        return abs(dep[x]-dep[t])+abs(dep[y]-dep[t]);
    }
    void solve(){
        cin>>n>>Q;
        init();
        for(int i=1;i<=n-1;i++){
            cin>>u>>v;
            add(u,v);
            add(v,u);
        }
        bfs();
        //for(int i=1;i<=n;i++) cout<<dep[i]<<" \n"[i==n];
        while(Q--){
            cin>>a>>b>>c>>d;
            int t1=lca(a,b);
            int t2=lca(c,d);
            //cout<<t1<<" "<<t2<<'\n';
            if(dis(a,t2)+dis(b,t2)==dis(a,b)
            ||dis(c,t1)+dis(d,t1)==dis(c,d)) cout<<"Y"<<'\n';
            else cout<<"N"<<'\n';
        }
    }
    int main(){
        ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
        int __=1;//cin>>__;
        while(__--)solve();return 0;
    }

[P3398 _ sugar - _ _ _ _luogu.com.cn]: https://www.luogu.com.cn/problem/P3398
[63205ab51fc33ab641ff3c43f9c8f4ed.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/61461b6556ea46858c3703f3b0da4d6a.png
[4cf0e5e55562545470e77604d20b476f.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/f61fa48cbbb74772b6dfc5e0d7798cb2.png