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;
}
每日一题题解 文章被收录于专栏

写题解

全部评论

相关推荐

10-28 10:48
已编辑
门头沟学院 Java
孩子我想要offer:发笔试后还没笔试把我挂了,然后邮箱一直让我测评没测,后面不知道干嘛又给我捞起来下轮笔试,做完测评笔试又挂了😅
点赞 评论 收藏
分享
从小父母离异家里没人管,靠着心里的不安和学校的环境也算是坚持到了学有所成的地步。到了大学环境开始松散不知道该做什么,只觉得在不挂科的基础上能往上考多少就考多少,等到秋招来临才发现自己有多么幼稚无能,今年九月份初才发现自己原来连一个求职的方向都没有。因为之前做过前后端一体的课设,算是有过了解,而对于其他岗位连做什么都不知道,因此这一个半个月在越来越焦虑的同时埋头苦学,事到如今想要活下去我似乎只能走前端这条路了,9月初先是靠着虚假夸大能力的简历得到一些笔试来确定了考察的方向,有一个大厂的无笔试面试最终是拒绝了没有勇气去面对。然后在这个基础上埋头苦学,如今也算是搭好了自己前端学习的框架和思考的瞄,可以逐渐给自己扩展新的知识和能力了,但这并不是一件多好的事儿,因为我发现学的越多越焦虑,学的越多便越无力。因为我感觉我如今努力学习的知识都是竞争对手们早就掌握了的东西,我如今困惑追求答案的难题早就被别人解决。别人早就能得心应手地做出项目而我连思考都会卡壳,看着别人的笔试和面经上那些闻所未闻的题目,我才知道别人到底有多强而我有多幼稚,我什么时候才能达到别人那种堪称熟练的能力呢?而且网上的焦虑越多越多,即便是真有这么高的能力最后也大概落得一个低薪打工人的下场,我真的感到迷茫。秋招都快结束了,而我还在继续痛苦的学习之旅,这些天找前端面试发现似乎问的有些简单跟网上搜到的内容不符(可能因为并不是大厂),我是不是本来就没打算被招所以别人懒得细问呢?我不知道,我只能继续总结下去学习下去,不管如何我都要活下去,如果我能早一些准备就好了,如果暑假能意识到现在这个情况就好了,可惜没有如果。种下一棵树的最好时间是十年前,其次是现在,虽然我相信自己的学习能力,但已经错过了最好的时机,只能在焦虑与痛苦中每天坚持学下去。目前的路还有很长很长,先去把typescript看了,再去巩固vue3的基础,再去练习elementui的使用,如果这能找到实习的话就好了。接下来呢?去学uniapp和小程序,不管如何我都要对得起曾经努力的自己。即便我们都感到痛苦,但我心中还是希望我们都能靠自己的努力来获取自己想要的幸福。
紧张的牛牛等一个of...:在担心什么呢,有一手985的学历在,就算是小厂别人都会要的,咱们双非的人更多,多少还在沉沦的,怕什么了
一句话证明你在找工作
点赞 评论 收藏
分享
评论
4
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务