BZOJ 1787: [Ahoi2008]Meet 紧急集合【LCA】

1787: [Ahoi2008]Meet 紧急集合

Time Limit: 20 Sec Memory Limit: 162 MB

Description

Input

Output

Sample Input

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

Sample Output

5 2
2 5
4 1
6 0

Source

Day1

题解

我们拿到题目，应该很容易就看出这是LCA吧，但是有三个量，这就有点问题了。
首先，我们知道，这个集合点肯定是三个最近公共祖先之一。
然后，我们想让线路重复的人不走，让另一个人走，这样就做到最优，也就是集合点是最近公共祖先中不相同的哪个点（肯定有两个相同），然后就解完了。

下面贴上代码：

#include<cstdio>
#include<iostream>
#define MAXN 500005
using namespace std;
int fa[MAXN],n,e,f[MAXN][31],dep[MAXN];
bool vis[MAXN];
struct xcw{
    int lnk[MAXN],nxt[2*MAXN],son[2*MAXN],tot;
    void add(int x,int y){son[++tot]=y;nxt[tot]=lnk[x];lnk[x]=tot;}
}a;
int read(){
    int ret=0;bool f=1;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar()) f^=!(ch^'-');
    for(;ch>='0'&&ch<='9';ch=getchar()) ret=ret*10+ch-48;
    return f?ret:-ret;
}
void DFS(int x,int deep){dep[x]=deep;for(int j=a.lnk[x];j;j=a.nxt[j]) if(!vis[a.son[j]]) vis[a.son[j]]=1,fa[a.son[j]]=x,DFS(a.son[j],deep+1);}
void chg(){
    for(int i=1;i<=n;i++){
        f[i][0]=fa[i];
        for(int j=1;(1<<j)<n;j++) f[i][j]=-1;
    }
    for(int j=1;(1<<j)<n;j++)
    for(int i=1,x;i<=n;i++) if(f[i][j-1]^-1) x=f[i][j-1],f[i][j]=f[x][j-1];
}
int get(int p,int q){
    int lg;
    if(dep[p]<dep[q]) swap(p,q);
    for(lg=1;(1<<lg)<=dep[p];lg++);lg--;
    for(int j=lg;j+1;j--)
    if(dep[p]-(1<<j)>=dep[q]) p=f[p][j];
    if(p==q) return p;
    for(int j=lg;j+1;j--)
    if(f[p][j]^-1&&f[p][j]^f[q][j]) p=f[p][j],q=f[q][j];
    return fa[p];
}
int cmp(int x,int y){int t=get(x,y);return dep[x]+dep[y]-dep[t]*2;}
int main(){
    n=read(),e=read();
    for(int i=1;i<n;i++){int x=read(),y=read();a.add(x,y);a.add(y,x);}
    vis[1]=1;fa[1]=1;
    DFS(1,0);chg();
    for(int i=1;i<=e;i++){
        int x=read(),y=read(),z=read();
        int ans,t,xy=get(x,y),yz=get(y,z),xz=get(x,z);//三个最近公共祖先 
        if(xy==xz) t=yz;else
        if(xy==yz) t=xz;else t=xy;
        ans=cmp(t,x)+cmp(t,y)+cmp(t,z);
        printf("%d %d\n",t,ans);
    }
    return 0;
}