[BZOJ1787][Ahoi2008]Meet 紧急集合

原题地址

倍增LCA复习题…

AC code:

#include <cstdio>
#include <vector>
using namespace std;
const int N=500010;
const int LOGN=25;
int n,m;
int d[N];
int f[N][LOGN];
vector<int> G[N];

void DFS(int x,int pre,int dep){
    f[x][0]=pre;d[x]=dep;
    for(int i=0;i<(int)G[x].size();i++){
        int y=G[x][i];
        if(y!=pre) DFS(y,x,dep+1);
    }
}

int lca(int x,int y){
    if(d[x]>d[y]) {int t=x;x=y;y=t;}
    while(d[x]!=d[y]){
        for(int i=0;;i++){
            if(d[f[y][i]]<d[x]){
                y=f[y][i-1];
                break;
            }
        }
    }
    if(x==y) return x;
    while(f[x][0]!=f[y][0]){
        for(int i=0;;i++){
            if(f[x][i]==f[y][i]){
                x=f[x][i-1];y=f[y][i-1];
                break;
            }
        }
    }
    return f[x][0];
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        G[u].push_back(v);G[v].push_back(u);
    }
    DFS(1,0,1);
    for(int i=1;i<LOGN;i++)
        for(int j=1;j<=n;j++)
            f[j][i]=f[f[j][i-1]][i-1];
    for(int i=1;i<=m;i++){
        int a,b,c,x,y,z,p,q,t,d1,d2;
        scanf("%d%d%d",&a,&b,&c);
        x=a;y=b;z=c;p=lca(a,b);
        t=q=lca(a,c);
        if(d[t]>d[p]){
            x=a;y=c;z=b;
            q=p;p=t;
        }
        t=lca(b,c);
        if(d[t]>d[p]){
            x=b;y=c;z=a;
            q=p;p=t;
        }
        d1=d[x]-d[p]+d[y]-d[p]+d[p]-d[q]+d[z]-d[q];
        d2=d[x]-d[q]+d[y]-d[q]+d[z]-d[q];
        if(d1<d2) printf("%d %d\n",p,d1);
        else printf("%d %d\n",q,d2);
    }

    return 0;
}