BZOJ 1787: [Ahoi2008]Meet 紧急集合【LCA】

Description

题解

集结地点肯定在三个点中某两点的LCA上，可以刷最小值。当然，通过观察，可以发现如果其中两个LCA相同，那么答案一定在另外一个。

代码

#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>#define maxn 500006using namespace std;inline char nc(){    static char buf[100000],*i=buf,*j=buf;    return i==j&&(j=(i=buf)+fread(buf,1,100000,stdin),i==j)?EOF:*i++;}inline int _read(){    char ch=nc();int sum=0;    while(!(ch>='0'&&ch<='9'))ch=nc();    while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();    return sum;}int n,T,tot,dep[maxn],lnk[maxn],nxt[maxn*2],son[maxn*2],f[maxn][19];bool vis[maxn];void add(int x,int y){    nxt[++tot]=lnk[x];son[tot]=y;lnk[x]=tot;}void dfs(int x){    vis[x]=0;    for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]])dep[son[j]]=dep[x]+1,f[son[j]][0]=x,dfs(son[j]);}void make_f(){    for(int j=1;j<=log2(n);j++)     for(int i=1;i<=n;i++) f[i][j]=f[f[i][j-1]][j-1];}int get(int x,int y){    if(dep[x]<dep[y])swap(x,y);    for(int j=log2(n);j>=0;j--) if(dep[x]-(1<<j)>=dep[y])x=f[x][j];    for(int j=log2(n);j>=0;j--) if(f[x][j]!=f[y][j])x=f[x][j],y=f[y][j];    if(x==y)return x;    return f[x][0];}int dis(int x,int y){int t=get(x,y);return dep[x]+dep[y]-2*dep[t];}int query(int x,int y,int z,int&P){    int p1=get(x,y),p2=get(x,z),p3=get(y,z);    if(p1==p2)P=p3;else    if(p1==p3)P=p2;else P=p1;    return dis(x,P)+dis(y,P)+dis(z,P);}int main(){    freopen("distance.in","r",stdin);    freopen("distance.out","w",stdout);    n=_read();T=_read();    for(int i=1,x,y;i<n;i++)x=_read(),y=_read(),add(x,y),add(y,x);    memset(vis,1,sizeof(vis));    dep[1]=1;dfs(1);make_f();    while(T--){        int x=_read(),y=_read(),z=_read(),P;        int t=query(x,y,z,P);        printf("%d %d\n",P,t);    }    return 0;}