【结论】【LCA】NKOJ 3773 紧急集合

NKOJ 3773 紧急集合
时间限制 : - MS 空间限制 : 165536 KB 该题不设总的时间（空间）限制，每组测试数据具有独立的时空限制
评测说明 : 2000ms
问题描述

样例输入

6 4
1 2
2 3
2 4
4 5
5 6
4 5 6
6 3 1
2 4 4
6 6 6

样例输出

5 2
2 5
4 1
6 0

结论：
树中到三个点距离和最小的一个点必在三个点两两的lca中。
若三个lca相等，则为这个lca。
若仅有两个相等，则为剩下一个lca。

代码：

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstdlib>
using namespace std;
const int need=500004;
const double lg2=log(2);

int n,m;
int fi[need],la[need<<1],en[need<<1];

int tot=0;
void add(int a,int b)
{
    tot++;
    la[tot]=fi[a];
    en[tot]=b;
    fi[a]=tot;
}

int dep[need];
int d=0;
bool vis[need];
int fa[need][22];

void dfs(int s)
{
    int k=ceil(log(dep[s])/lg2);
    d=max(d,k);
    for(int i=1;i<=k;i++) fa[s][i]=fa[fa[s][i-1]][i-1];
    for(int t=fi[s],y;t;t=la[t])
    {
        y=en[t];
        if(vis[y]) continue;
        vis[y]=true;
        dep[y]=dep[s]+1;
        fa[y][0]=s;
        dfs(y);
    }
}

int lca(int v,int u)
{
    if(dep[u]>dep[v]) swap(u,v);
    int p=dep[v]-dep[u];
    for(int i=0;i<=d;i++) if((p>>i)&1) v=fa[v][i];
    if(u==v) return v;
    for(int i=d;i>=0;i--) if(fa[u][i]!=fa[v][i]) v=fa[v][i],u=fa[u][i];
    return fa[u][0];
}

int getdis(int a,int b)
{
    int c=lca(a,b);
    return dep[a]-dep[c]+dep[b]-dep[c];
}

int work(int a,int b,int c,int e)
{

    return getdis(a,e)+getdis(b,e)+getdis(c,e);
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1,a,b;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    dfs(1);
    for(int i=1,a,b,c,f[3],e;i<=m;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        f[0]=lca(a,b),f[1]=lca(b,c),f[2]=lca(a,c);
        if(f[0]==f[1]) e=f[2];
        else if(f[0]==f[2]) e=f[1];
        else if(f[1]==f[2]) e=f[0];
        printf("%d %d\n",e,work(a,b,c,e));
    }
}