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

Description

题解

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

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 500006
using 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;
}