codeforces 519E 倍增法LCA

点击打开链接

题意：给一个树，然后每次询问需要统计树上的点到达a和b的最近距离相等，求这样的点的个数

思路：因为是一个树，所以最近距离便是LCA上的边，那么就可以看出来当a和b的距离是奇数时结果为0，当是偶数时，最中间的那个点可以，并且与它相连的不是LCA上的点都是可以的，自己画一个图很好看出来      以前的模版不能快速求出某个点的第几个父亲，而这个可以，以后就用这个，其实记录父节点暴力上去也可以过，慢了好多

#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=100010;
vector<int>G[maxn];
int dis[maxn],lca[maxn][20],f[maxn],sum[maxn];
void dfs(int x,int fa){
    dis[x]=dis[fa]+1;f[x]=fa;
    lca[x][0]=fa;
    for(int i=1;i<=19;i++){
        if(lca[x][i-1]==-1) break;
        lca[x][i]=lca[lca[x][i-1]][i-1];
    }
    for(int i=0;i<G[x].size();i++){
        int t=G[x][i];
        if(t==fa) continue;
        dfs(t,x);sum[x]+=sum[t];
    }
}
int findlca(int x,int y){
    if(dis[x]<dis[y]) swap(x,y);
    for(int i=19;i>=0;i--){
        if(lca[x][i]!=-1&&dis[lca[x][i]]>=dis[y]) x=lca[x][i];
    }
    if(x==y) return x;
    for(int i=19;i>=0;i--){
        if(lca[x][i]-lca[y][i]){
            x=lca[x][i];y=lca[y][i];
        }
    }
    return lca[x][0];
}
int findkth(int x,int k){
    for(int i=19;i>=0;i--){
        if((1<<i)<=k){
            x=lca[x][i];
            k-=(1<<i);
        }
    }
    return x;
}
int main(){
    int n,m,a,b;
    while(scanf("%d",&n)!=-1){
        memset(dis,0,sizeof(dis));
        for(int i=0;i<maxn;i++) G[i].clear(),sum[i]=1;
        for(int i=0;i<n-1;i++){
            scanf("%d%d",&a,&b);
            G[a].push_back(b);
            G[b].push_back(a);
        }
        dfs(1,0);
        scanf("%d",&m);
        while(m--){
            scanf("%d%d",&a,&b);
            if(a==b){
                printf("%d\n",n);continue;
            }
            int fa=findlca(a,b);
            int len=dis[a]+dis[b]-2*dis[fa];
            if(len%2==1){printf("0\n");continue;}
            if(dis[a]-dis[fa]==len/2){
                int ff1=findkth(a,len/2-1);
                int ff2=findkth(b,len/2-1);
                printf("%d\n",n-sum[ff1]-sum[ff2]);continue;
            }
            if(dis[a]-dis[fa]>len/2){
                int fff=findkth(a,len/2-1);
                printf("%d\n",sum[f[fff]]-sum[fff]);
            }else{
                int fff=findkth(b,len/2-1);
                printf("%d\n",sum[f[fff]]-sum[fff]);
            }
        }
    }
    return 0;
}