【LCA】BZOJ1832 & BZOJ1787(AHOI)-集会

【题目大意】

一个图有n个点n-1条边（也就是说是一棵树），求其中三点共同到达某一点经过总共的最少边数以及共同到达的那一点。

【思路】

借用一下黄学长给的结论：三个点两两取LCA，其中必有两个相同，则另外一个点就是答案。

注意BZOJ大数据要用scanf和printf，因为cout的原因RE了好几次_(:зゝ∠)_基本算是一道标程类的题目了啦。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
const int MAXN=500001;
const int M=20;
using namespace std;
int n,m;
vector<int> G[MAXN];
int fa[MAXN][M];
int dep[MAXN];

void dfs(int v,int p,int d)
{
    fa[v][0]=p;
    dep[v]=d;
    for (int i=0;i<G[v].size();i++)
        if (G[v][i]!=p) dfs(G[v][i],v,d+1);
}

void kb()
{
    for (int i=1;i<M;i++)
        for (int j=1;j<=n;j++)
        {
            fa[j][i]=fa[fa[j][i-1]][i-1];
        }
}

void init()
{
    scanf("%d%d",&n,&m);
    for (int i=0;i<n-1;i++)
    {
        int fr,to;
        scanf("%d%d",&fr,&to);
        G[fr].push_back(to);
        G[to].push_back(fr);
    }
    dfs(1,0,0);
    kb();
}

int swim(int x,int H)
{
    for (int i=0;H>0;i++)
    {
        if (H&1) x=fa[x][i];
        H/=2;
    }
    return x;
}

int LCA(int u,int v)
{
    if (dep[u]<dep[v]) swap(u,v);
    u=swim(u,dep[u]-dep[v]);
    if (u==v) return u;
    for (int i=M-1;i>=0;i--)
    {
        if (fa[u][i]!=fa[v][i])
        {
            u=fa[u][i];
            v=fa[v][i];
        }
    }
    return fa[u][0];
}

int dis(int x,int y)
{
    int z=LCA(x,y);
    int re=dep[x]+dep[y]-2*dep[z];
    return re;
}

void query()
{
    int u,v,w;
    for (int i=0;i<m;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        int uv=LCA(u,v);
        int uw=LCA(u,w);
        int vw=LCA(v,w);
        int pos;
        if (uv==uw) pos=vw;
            else if (uv==vw) pos=uw;
                else pos=uv;
         int cost=dis(u,pos)+dis(v,pos)+dis(w,pos);
         printf("%d %d
",pos,cost);
    }
}

int main()
{
    init();
    query();
    return 0;
}