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

题目描述

传送门

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

题解

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

代码

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#define N 1000003using 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);    }}