cf519E. A and B and Lecture Rooms

链接

点击跳转

题解

如果两个点的距离是奇数，无解

如果两个点的距离是偶数，那么肯定路径的中点满足条件，然后想象以这个中点为根建树，只要接下来不走到$a$或$b$所在的子树就可以

代码

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define iinf 0x3f3f3f3f
#define linf (1ll<<60)
#define eps 1e-8
#define maxn 1000010
#define maxe 1000010
#define maxk 20
#define cl(x) memset(x,0,sizeof(x))
#define rep(_,__) for(_=1;_<=(__);_++)
#define em(x) emplace(x)
#define emb(x) emplace_back(x)
#define emf(x) emplace_front(x)
#define fi first
#define se second
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll read(ll x=0)
{
    ll c, f(1);
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
    for(;isdigit(c);c=getchar())x=x*10+c-0x30;
    return f*x;
}
struct Graph
{
    int etot, head[maxn], to[maxe], next[maxe], w[maxe];
    void clear(int N)
    {
        for(int i=1;i<=N;i++)head[i]=0;
        etot=0;
    }
    void adde(int a, int b, int c=0){to[++etot]=b;w[etot]=c;next[etot]=head[a];head[a]=etot;}
}G;
struct Doubling_LCA
{
    int f[maxn][maxk+2], depth[maxn], size[maxn];
    void clear(int n){for(int i=1;i<=n;i++)depth[i]=0, cl(f[i]);}
    void dfs(Graph &G, int pos, int pre)
    {
        size[pos]=1;
        for(auto k=1;(1<<k)<=depth[pos];k++)f[pos][k]=f[f[pos][k-1]][k-1];
        for(auto p(G.head[pos]);p;p=G.next[p])
            if(G.to[p]!=pre)
            {
                f[G.to[p]][0]=pos;
                depth[G.to[p]]=depth[pos]+1;
                dfs(G,G.to[p],pos);
                size[pos]+=size[G.to[p]];
            }
    }
    void run(Graph &G, int root)
    {
        depth[root]=1;
        dfs(G,root,0);
    }
    int q(int x, int y)
    {
        if(depth[x]<depth[y])swap(x,y);
        for(auto k(maxk);~k;k--)
            if(depth[f[x][k]]>=depth[y])
                x=f[x][k];
        if(x==y)return x;
        for(auto k(maxk);~k;k--)
            if(f[x][k]!=f[y][k])
                x=f[x][k], y=f[y][k];
        return f[x][0];
    }
}db;
ll jp(ll x, ll step)
{
    for(ll k=maxk;~k;k--)if(step&(1ll<<k))x=db.f[x][k];
    return x;
}
ll n, a, b, ans, q;
int main()
{
    ll i;
    n=read();
    rep(i,n-1)
    {
        a=read(), b=read();
        G.adde(a,b), G.adde(b,a);
    }
    db.run(G,1);
    q=read();
    while(q--)
    {
        a=read(), b=read();
        ll lca=db.q(a,b), dis=db.depth[a]+db.depth[b]-2*db.depth[lca];
        if(dis&1)
        {
            printf("0\n");
        }
        else
        {
            if(db.depth[a]==db.depth[b])
            {
                if(a==b)
                {
                    printf("%lld\n",n);
                }
                else
                {
                    ll x=jp(a,dis/2-1), y=jp(b,dis/2-1);
                    printf("%lld\n",n-db.size[x]-db.size[y]);
                }
            }
            else
            {
                ll x, ans;
                if(db.depth[a]>db.depth[b])x=a;
                else x=b;
                ans=n-db.size[jp(x,dis/2-1)];
                ans-=n-db.size[jp(x,dis/2)];
                printf("%lld\n",ans);
            }
        }
    }
    return 0;
}