题意:给一个树,然后每次询问需要统计树上的点到达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;
}