|
问题描述
现有一棵 n 个节点的树,树上每条边的长度均为 1。给出 m 个询问,每次询问两个节 点 x,y,求树上到 x,y 两个点距离相同的节点数量。
输入格式
第一个整数 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
思路: 被坑了好久 一直没调出来 复习了一波LCA 后 终于A 了 注意 倍增的用法 即GOUP 操作
顺便复习一下 距离倍增
code:
g[x][i]=g[x][i-1]+g[fa[x][i-1]][i-1];
dis 求法
for(i=0;i<=ss;i++) if(k&(1<<i))dis+=g[x][i],x=fa[x][i]; if(x==y)return dis; for(i=s;i>=0;i--) if(fa[x][i]!=fa[y][i]) { dis+=g[x][i];x=fa[x][i]; dis+=g[y][i];y=fa[y][i]; } return dis+=g[x][0]+g[y][0];
code:
// #include<bits/stdc++.h> using namespace std; #define maxnn 600000 int f[maxnn][100]; int las[maxnn],nex[maxnn],en[maxnn],tot; int n,m; int dep[maxnn]; int x,y; int size[maxnn]; void dfs(int v,int fa) { dep[v]=dep[fa]+1; f[v][0]=fa; int s=ceil(log2(n)); for(int i=1;i<=s;i++) { f[v][i]=f[f[v][i-1]][i-1]; } for(int i=las[v];i;i=nex[i]) { int u=en[i]; if(u!=fa) { dfs(u,v); size[v]+=size[u]; } } return ; } void add(int a,int b) { en[++tot]=b; nex[tot]=las[a]; las[a]=tot; } int goup(int x,int s) { int k=ceil(log2(n)); for(int i=0;i<=k;i++) { if((s&(1<<i))) { x=f[x][i]; } } return x; } int lca(int x,int y) { if(dep[x]<dep[y])swap(x,y); x=goup(x,dep[x]-dep[y]); if(x==y) return y; int s=ceil(log2(dep[x])); for(int i=s;i>=0;i--) { if(f[x][i]!=f[y][i]) { x=f[x][i]; y=f[y][i]; } } return f[x][0]; } int main() { cin>>n; for(int i=1;i<=n;i++) size[i]=1; for(int i=1;i<n;i++) { scanf("%d%d",&x,&y); add(x,y); add(y,x); } dfs(1,0); scanf("%d",&m); for(int i=1;i<=m;i++) { int k1=0,k=0; scanf("%d%d",&x,&y); if(y==x) {cout<<n<<endl;continue;} if(dep[x]==dep[y]) { int k=dep[x]-dep[lca(x,y)]; int k1=n-size[goup(x,k-1)]-size[goup(y,k-1)]; cout<<k1<<endl; continue; } if((abs((dep[x]-dep[y])))&1) {cout<<0<<endl;continue;} if(dep[x]<dep[y]) swap(x,y); k=(dep[y]+dep[x]-2*dep[lca(x,y)])/2; int p1=goup(x,k-1);int p2=goup(x,k); k1=size[p2]-size[p1]; cout<<k1<<endl; } }
原文地址:https://www.cnblogs.com/OIEREDSION/p/11324624.html
时间: 2024-10-13 21:41:54