【结论】【树(LCA)】NKOJ3815 树上的询问

NKOJ3815 树上的询问
时间限制 : - MS 空间限制 : 265536 KB
评测说明 : 1000ms

问题描述
现有一棵 n 个节点的树，树上每条边的长度均为 1。给出 m 个询问，每次询问两个节 点 a,b，求树上到 a,b两个点距离相同的节点数量。

输入格式
第一个整数 n，表示树有 n 个点。
接下来 n-1 行每行两整数 a，b，表示从 a 到 b 有一条边。
接下来一行一个整数 m，表示有 m 个询问。
接下来 m 行每行两整数 x，y，询问到 x 和 y 距离相同的点的数量。

输出格式
共 m 行，每行一个整数表示询问的答案。

样例输入
7
1 2
1 3
2 4
2 5
3 6
3 7
3
1 2
4 5
2 3

样例输出
0
5
1

提示
【数据规模】
30%的数据，满足 n≤50，m≤50 对于
60%的数据，满足 n≤1000，m≤1000 对于
100%的数据，满足 n≤100000，m≤100000

观察发现：

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstdlib>
using namespace std;
const int need=100003;
const double lg2=log(2);

int n,m;
//.............................................................
inline void in_(int &d)
{
    char t=getchar();
    while(t<'0'||t>'9') t=getchar();
    for(d=0;!(t<'0'||t>'9');t=getchar()) d=(d<<1)+(d<<3)+t-'0';
}
//.............................................................
int fi[need],la[need<<1],en[need<<1];

int tot=0;
void add(int a,int b)
{
    tot++;
    la[tot]=fi[a];
    en[tot]=b;
    fi[a]=tot;
}
//.............................................................
int dep[need];
int d=0;
bool vis[need];
int fa[need][22],son[need];

void dfs(int s)
{
    vis[s]=true;
    int k=ceil(log(dep[s])/lg2);
    d=max(d,k);
    for(int i=1;i<=k;i++) fa[s][i]=fa[fa[s][i-1]][i-1];
    for(int t=fi[s],y;t;t=la[t])
    {
        y=en[t];
        if(vis[y]) continue;
        dep[y]=dep[s]+1;
        fa[y][0]=s;
        dfs(y);
        son[s]+=son[y]+1;
    }
}

int go_up(int v,int p)
{
    for(int i=0;i<=d;i++) if((p>>i)&1) v=fa[v][i];
    return v;
}

int lca(int v,int u)
{
    if(dep[u]>dep[v]) swap(u,v);
    v=go_up(v,dep[v]-dep[u]);
    if(u==v) return v;
    for(int i=d;i>=0;i--) if(fa[u][i]!=fa[v][i]) v=fa[v][i],u=fa[u][i];
    return fa[u][0];
}


int getdis(int a,int b)
{
    int c=lca(a,b);
    return dep[a]+dep[b]-dep[c]-dep[c];
}
int getdis(int a,int b,int c)
{
    return dep[a]+dep[b]-dep[c]-dep[c];
}
//.............................................................
int main()
{
    scanf("%d",&n);
    for(int i=1,a,b;i<n;i++)
    {
        in_(a),in_(b);//scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    dfs(1);

    scanf("%d",&m);
    for(int i=1,a,b,c,a1,ab,ac,bc,ans;i<=m;i++)
    {
        in_(a),in_(b);//scanf("%d%d",&a,&b);
        c=lca(a,b);
        ab=getdis(a,b,c);
        if(ab&1) ans=0;
        else
        {
            if(a==b) ans=n;
            else
            {
                ac=getdis(a,c);
                bc=getdis(b,c);
                if(ac==bc) 
                {
                    a=go_up(a,dep[a]-dep[c]-1);
                    b=go_up(b,dep[b]-dep[c]-1);
                    ans=n-(son[a]+1)-(son[b]+1);
                }
                else
                {
                    if(dep[b]>dep[a]) swap(a,b);
                    a=go_up(a,ab/2-1);
                    a1=go_up(a,1);
                    ans=son[a1]+1-(son[a]+1);
                }
            }
        }
        printf("%d\n",ans);
    }
}