BZOJ 1832: [AHOI2008]聚会

1832: [AHOI2008]聚会

Description

Y岛风景美丽宜人，气候温和，物产丰富。Y岛上有N个城市，有N-1条城市间的道路连接着它们。每一条道路都连接某两个城市。幸运的是，小可可通过这些道路可以走遍Y岛的所有城市。神奇的是，乘车经过每条道路所需要的费用都是一样的。小可可，小卡卡和小YY经常想聚会，每次聚会，他们都会选择一个城市，使得3个人到达这个城市的总费用最小。 由于他们计划中还会有很多次聚会，每次都选择一个地点是很烦人的事情，所以他们决定把这件事情交给你来完成。他们会提供给你地图以及若干次聚会前他们所处的位置，希望你为他们的每一次聚会选择一个合适的地点。
Input

第一行两个正整数，N和M。分别表示城市个数和聚会次数。后面有N-1行，每行用两个正整数A和B表示编号为A和编号为B的城市之间有一条路。城市的编号是从1到N的。再后面有M行，每行用三个正整数表示一次聚会的情况：小可可所在的城市编号，小卡卡所在的城市编号以及小YY所在的城市编号。
Output

一共有M行，每行两个数Pos和Cost，用一个空格隔开。表示第i次聚会的地点选择在编号为Pos的城市，总共的费用是经过Cost条道路所花费的费用。
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

数据范围：

100%的数据中，N<=500000，M<=500000。

40%的数据中N<=2000，M<=2000。

题解：
这题 要是换成两个人 就是简单的lca裸体
三个人其实也一样
我们可以三个人两两求lca得到的三个lca
答案就是 其中不重复的然后再算距离就会了。

#include<bits/stdc++.h>
using namespace std;
const int N = 500007;
int head[N], top = 1;

struct EDGE{
    int v, next;
} e[2 * N];

int n, m;

void add_edge(int u,int v){
    e[top].v = v;
    e[top].next = head[u];
    head[u] = top++;
}

int size[N], son[N], deep[N],up[N],father[N];

void dfs1(int u,int fa){
    size[u] = 1;
    deep[u] = deep[fa] + 1;
    for (int i = head[u]; i;i=e[i].next){
        int to = e[i].v;
        if(to!=fa){
            dfs1(to, u);
            size[u] += size[to];
            if(size[son[u]]<size[to]){
                son[u] = to;
            }
        }
    }
}

void dfs2(int u,int fa,int p){
    father[u] = fa;
    up[u] = p;
    if(son[u]){
        dfs2(son[u], u, p);
    }
    for (int i = head[u]; i;i=e[i].next){
        int to = e[i].v;
        if(to!=fa&&to!=son[u]){
            dfs2(to, u, to);
        }
    }
}

int lca(int x,int y){
    while(up[x]!=up[y]){
        if(deep[up[x]]>deep[up[y]]){
            x = father[up[x]];
        }else{
            y = father[up[y]];
        }
    }
    if(deep[x]>deep[y]){
        return y;
    }
    return x;
}

int main(){
    scanf("%d%d", &n, &m);
    for (int i = 1; i < n;i++){
        int u, v;
        scanf("%d%d", &u, &v);
        add_edge(u, v);
        add_edge(v, u);
    }
    dfs1(1, 0);
    dfs2(1, 0, 0);
    while(m--){
        int x, y,z;
        scanf("%d%d%d", &x, &y,&z);
        int lca1 = lca(x, y);
        int lca2 = lca(x, z);
        int lca3 = lca(z, y);
        int pos, ans;
        if(lca1==lca2){
            pos = lca3;
        }else if(lca2==lca3){
            pos = lca1;
        }else{
            pos = lca2;
        }
        ans = deep[x] + deep[y] + deep[z] - deep[lca1] - deep[lca2] - deep[lca3];
        printf("%d %d\n", pos, ans);
    }

}