cf519E

传送门

多组询问，问到树上两个点x,y距离相等的点的个数。

倍增求lca.

//Twenty
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
#include<ctime>
const int maxn=1e5+;
typedef long long LL;
using namespace std;
int n,m,ecnt,fir[maxn],nxt[maxn<<],to[maxn<<],f[maxn][],R[maxn],sz[maxn];;
void add(int u,int v) {
    nxt[++ecnt]=fir[u]; fir[u]=ecnt; to[ecnt]=v;
    nxt[++ecnt]=fir[v]; fir[v]=ecnt; to[ecnt]=u;
}
void dfs(int x,int fa) {
    R[x]=R[fa]+;
    f[x][]=fa;
    sz[x]=;
    for(int i=fir[x];i;i=nxt[i]) if(to[i]!=fa) {
        dfs(to[i],x);
        sz[x]+=sz[to[i]];
    }
}
void make_st() {
    for(int i=;i<=;i++)
        for(int j=;j<=n;j++)
            f[j][i]=f[f[j][i-]][i-];
}
int lca(int x,int y) {
    if(R[x]<R[y]) swap(x,y);
    for(int i=;i>=;i--)
        if(R[f[x][i]]>=R[y]) x=f[x][i];
    if(x==y) return x;
    for(int i=;i>=;i--)
        if(f[x][i]!=f[y][i]) {
            x=f[x][i];
            y=f[y][i];
        }
    return f[x][];
}
int cal(int x,int y) {
    if(x==y) return sz[];
    int z=lca(x,y);
    if(R[x]==R[y]) {
        for(int i=;i>=;i--)
            if(f[x][i]!=f[y][i]) {
            x=f[x][i];
            y=f[y][i];
            }
        return sz[]-sz[x]-sz[y];
    }
    int l=R[x]-R[z]+R[y]-R[z];
    if(l&) return ; l>>=;
    if(R[x]<R[y]) swap(x,y);
    l=R[x]-R[z]-l+;
    for(int i=;i>=;i--)
        if(R[f[x][i]]&&R[f[x][i]]-R[z]>=l)
            x=f[x][i];
    return sz[f[x][]]-sz[x];
}
int main()
{
    scanf("%d",&n);
    for(int i=;i<n;i++) {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }
    dfs(,);
    make_st();
    scanf("%d",&m);
    for(int i=;i<=m;i++) {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",cal(x,y));
    }
    return ;
}