A and B and Lecture Rooms

A and B and Lecture Rooms

https://ac.nowcoder.com/acm/problem/110856

题意:
有一颗n个节点的树,有m个询问,每个询问给出x和y两个节点,让你求树上节点到这两个节点距离相等的数目。

思路:
①:如果x和y深度相同,则他们的最近公共祖先lca(x,y)的非这二个节点方向的节点以外的节点到x和y的距离相等.
②:如果x和深度不同,则他们的路径为奇数时无解,为偶数时中点在深度大的一方,该点到x和y的方向节点以外的节点到x和y的距离相等.

代码:

#include<bits/stdc++.h>
#define ll long long
using namespace std;

int read()
{
    int x=0, g=1;
    char c=getchar();
    while(c<'0'||c>'9')
    {
        if(c=='-')
        {
            g=-1;
        }
        c=getchar();
    }
    while(c<='9'&&c>='0')
    {
        x=(x<<3)+(x<<1)+(c^48);
        c=getchar();
    }
    return x*g;
}

int n, q, parent[22][100005], dep[100005], se[100005];
vector<int> g[100005];

int dfs(int v,int d, int f)
{
    dep[v]=d;
    parent[0][v]=f;
    se[v]=1;
    for(int i=0; i<g[v].size(); i++)
    {
        if(g[v][i]!=f)
            se[v]+=dfs(g[v][i],d+1,v);
    }
    return se[v];
}

int lca(int u,int v)
{
    if(dep[u]<dep[v])
    {
        swap(u,v);
    }
    for(int i=21;i>=0;i--)
    {
        if((dep[u]-dep[v])>=(1<<i))
        {
            u=parent[i][u];
        }
    }
    if(u==v)
    {
        return u;
    }
    for(int i=21;i>=0;i--)
    {
        if(parent[i][u]!=parent[i][v])
        {
            u=parent[i][u];
            v=parent[i][v];
        }
    }
    return parent[0][u];
}

int shang(int v,int d)
{
    for(int i=21;i>=0;i--)
    {
        if((d>>i)&1)
        {
            v=parent[i][v];
        }
    }
    return v;
}

int fun(int x,int y)
{
    if(dep[x]==dep[y])
    {
        if(x==y)
        {
            return n;
        }
        int v=lca(x,y);
        int d=dep[x]-dep[v];
        return n-se[shang(x,d-1)]-se[shang(y,d-1)];
    }
    else
    {
        int v=lca(x,y);
        int dx=dep[x]-dep[v], dy=dep[y]-dep[v];
        if(dx>dy)
        {
            int d=dx-dy;
            if(d%2==0)
            {
                int ki=shang(x,dx-d/2-1);
                return se[parent[0][ki]]-se[ki];
            }
            else
            {
                return 0;
            }
        }
        else
        {
            int d=dy-dx;
            if(d%2==0)
            {
                int ki=shang(y,dy-d/2-1);
                return se[parent[0][ki]]-se[ki];
            }
            else
            {
                return 0;
            }
        }
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=0; i<n-1; i++)
    {
        int u, v;
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    memset(parent,-1,sizeof(parent));
    dfs(1,0,-1);
    for(int i=1;i<21;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(parent[i-1][j]<1)
            {
                parent[i][j]=-1;
            }
            else
            {
                parent[i][j]=parent[i-1][parent[i-1][j]];
            }
        }
    }
    scanf("%d",&q);
    while(q--)
    {
        int x, y;
        scanf("%d%d",&x,&y);
        cout << fun(x,y) << endl;
    }
    return 0;
}
每日一题题解 文章被收录于专栏

写题解

全部评论

相关推荐

ProMonkey2024:5个oc?厉害! 但是有一个小问题:谁问你了?😡我的意思是,谁在意?我告诉你,根本没人问你,在我们之中0人问了你,我把所有问你的人都请来 party 了,到场人数是0个人,誰问你了?WHO ASKED?谁问汝矣?誰があなたに聞きましたか?누가 물어봤어?我爬上了珠穆朗玛峰也没找到谁问你了,我刚刚潜入了世界上最大的射电望远镜也没开到那个问你的人的盒,在找到谁问你之前我连癌症的解药都发明了出来,我开了最大距离渲染也没找到谁问你了我活在这个被辐射蹂躏了多年的破碎世界的坟墓里目睹全球核战争把人类文明毁灭也没见到谁问你了(别的帖子偷来的,现学现卖😋)
点赞 评论 收藏
分享
4 收藏 评论
分享
牛客网
牛客企业服务