bzoj 1787: [Ahoi2008]Meet 紧急集合 （LCA）

题目描述

传送门

题目大意：给出一棵树，每个询问包含3个点。找出一点使三个点到这个点的距离和最小。

题解

答案一定在某两个点的lca上。。。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000003
using namespace std;
int point[N],v[N],nxt[N],fa[N][20],mi[20];
int n,m,tot,deep[N],ans,mn,ansx;
void add(int x,int y)
{
    tot++; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    tot++; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int f)
{
    deep[x]=deep[f]+1;
    for (int i=1;i<=19;i++) {
        if (deep-mi[i]<0) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }
    for (int i=point[x];i;i=nxt[i]){
        if (v[i]==f) continue;
        fa[v[i]][0]=x;
        dfs(v[i],x);
    }
}
int lca(int x,int y)
{
    if (deep[x]<deep[y]) swap(x,y);
    int k=deep[x]-deep[y];
    for (int i=0;i<=19;i++)
     if ((k>>i)&1) x=fa[x][i],ans+=mi[i];
    if (x==y) return x;
    for (int i=19;i>=0;i--)
     if (fa[x][i]!=fa[y][i])
      x=fa[x][i],y=fa[y][i],ans+=mi[i]*2;
    ans+=2;
    return fa[x][0];
}
void solve(int x,int y,int z)
{
    int sum=0;
    ans=0;
    int t=lca(x,y); sum+=ans;
    ans=0; lca(t,z);
    sum+=ans;
    if (sum<mn) mn=sum,ansx=t;
}
int main()
{
    freopen("a.in","r",stdin);
    scanf("%d%d",&n,&m); mi[0]=1;
    for (int i=1;i<=19;i++) mi[i]=mi[i-1]*2;
    for (int i=1;i<n;i++) {
        int x,y; scanf("%d%d",&x,&y);
        add(x,y);
    }
    dfs(1,0);
    for (int i=1;i<=m;i++){
        int x,y,z; scanf("%d%d%d",&x,&y,&z);
        mn=n*3;
        solve(x,y,z);
        solve(x,z,y);
        solve(y,z,x);
        printf("%d %d\n",ansx,mn);
    }
}