bzoj 1832: [AHOI2008]聚会 lca

来源：互联网发布：数据泄露防护系统编辑：程序博客网时间：2024/05/17 05:51

题意：给出一棵树，每条边的权值为1,。每次询问给出三个点，求一个点使得这三个点到该点的距离和最小。

分析：求出三个点两两之间的lca，容易得知肯定有两个是相同的，那么我们选取不同的那个就好了。

代码：

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>#define N 500005using namespace std;int n,m,cnt,dep[N],fa[N][17],last[N];struct edge{int to,next;}e[N*2];void insert(int u,int v){e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;}void dfs(int x){for (int i=1;i<=16;i++)if (fa[x][i-1]) fa[x][i]=fa[fa[x][i-1]][i-1];for (int i=last[x];i;i=e[i].next){if (e[i].to==fa[x][0]) continue;dep[e[i].to]=dep[x]+1;fa[e[i].to][0]=x;dfs(e[i].to);}}int lca(int x,int y){if (dep[x]<dep[y]) swap(x,y);for (int i=16;i>=0;i--)if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];if (x==y) return x;for (int i=16;i>=0;i--)if (fa[x][i]!=fa[y][i]) {x=fa[x][i];y=fa[y][i];}return fa[x][0];}int main(){scanf("%d%d",&n,&m);for (int i=1;i<n;i++){int u,v;scanf("%d%d",&u,&v);insert(u,v);}dep[1]=1;dfs(1);for (int i=1;i<=m;i++){int x,y,z;scanf("%d%d%d",&x,&y,&z);int x1=lca(x,y),y1=lca(x,z),z1=lca(y,z);if (x1==y1) printf("%d %d\n",z1,dep[y]+dep[z]-dep[z1]*2+dep[x]+dep[z1]-dep[x1]*2);else if (x1==z1) printf("%d %d\n",y1,dep[x]+dep[z]-dep[y1]*2+dep[y]+dep[y1]-dep[x1]*2);else printf("%d %d\n",x1,dep[x]+dep[y]-dep[x1]*2+dep[x1]+dep[z]-dep[y1]*2);}return 0;}

0 0