[codefoces519E]A and B and Lecture Rooms(lca)

题目:

我是超链接

题意:

求树上到某两点距离相等的点有多少个。

题解:

容易发现,两个点之间如果有偶数个点ans=0;如果是奇数个,ans=中间点排除这两条分支的链长
他俩的中间点不是公共父亲就是二者上下差距的中点,讨论一下

代码:

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#define N 100005
#define sz 19
using namespace std;
int tot,nxt[N*2],point[N],v[N*2],h[N],f[N][sz],dis[N],mi[sz],x,y,z,size[N];
void addline(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int fa)
{
    h[x]=h[fa]+1;size[x]=1;
    for (int i=1;i<sz;i++)
      if (h[x]<mi[i]) break;
      else f[x][i]=f[f[x][i-1]][i-1];
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa)
      {
        f[v[i]][0]=x;
        dis[v[i]]=dis[x]+1;
        dfs(v[i],x);
        size[x]+=size[v[i]];
      }
}
int lca(int x,int y)
{
    if (h[x]<h[y]) swap(x,y);
    int k=h[x]-h[y];
    for (int i=0;i<sz;i++)
      if ((k>>i)&1) x=f[x][i];
    if (x==y) return x;
    for (int i=sz-1;i>=0;i--)
      if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
int jump(int x,int k)
{
    for (int i=sz-1;i>=0;i--)
      if (h[f[x][i]]>k) x=f[x][i];
    return x;
}
int main()
{
    int i,n,m;
    mi[0]=1;
    for (i=1;i<sz;i++) mi[i]=mi[i-1]*2;
    scanf("%d",&n);
    for (i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addline(x,y);
    }
    scanf("%d",&m);
    dfs(1,0);
    while (m--)
    {
        scanf("%d%d",&x,&y);
        if (x==y){printf("%d\n",n);continue;} 
        int fa=lca(x,y);
        int diss=dis[x]+dis[y]-dis[fa]*2;
        if (diss%2==1){printf("0\n");continue;}
        diss/=2;
        if (h[x]-diss==h[fa]) 
        {
            int pre1=jump(x,h[fa]);
            int pre2=jump(y,h[fa]);
            printf("%d\n",n-size[pre1]-size[pre2]);
        }
        else
        {
            if (h[x]<h[y]) swap(x,y);
            int pre=jump(x,h[x]-diss);
            z=f[pre][0]; 
            printf("%d\n",size[z]-size[pre]);
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值