【ACWing】391. 聚会

题目地址：

https://www.acwing.com/problem/content/description/393/

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

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

输出格式：
一共有$M$行，每行两个数Pos和Cost，用一个空格隔开，表示第$i$次聚会的地点选择在编号为Pos的城市，总共的费用是经过Cost条道路所花费的费用。

数据范围：
$N\le 500000,M\le 500000$

先设点$1$为树根，做一下最近公共祖先的预处理，参考https://blog.csdn.net/qq_46105170/article/details/116217633。

问题即为，给定树中$3$个点$a,b,c$，求树上的一个点$x$，使得三个点与$x$的距离之和最小。取$1$为树根，对这三个点两两求最近公共祖先。首先证明，这三个最近公共祖先必有两个是同一个点，并且如果三个最近公共祖先不都重合，则两个不同的最近公共祖先深度一定不同，并且三个点的最近公共祖先就是深度小的那个最近公共祖先。那么这三个点有下面几种可能：
1、某一个点是树根$1$，那最近公共祖先就是$1$，成立；
2、三个点在三个不同子树中，那$1$是最近公共祖先，三个最近公共祖先都是$1$；
3、某两个点在同一棵子树中，另一个点在另一棵子树中，则这个点与另外两个点的最近公共祖先都是$1$，并且那另外两个点的最近公共祖先在它们所在的子树中；
4、三个点在同一棵子树中，只考虑那棵子树，由归纳假设，命题成立。

我们考虑两个不同的最近公共祖先的情况，设为$x,y$，并设$x$深度大，$y$深度小。由于$y$是三个点的最近公共祖先，答案一定在$y$为树根的子树里（选别的点总距离一定不会更优）；进一步地，答案一定在$x$为树根的子树里（否则子树之外的那个点来聚会的话，不如选$x$总距离小），从而$x$就是答案（因为$x$子树的点都不如$x$更优）。如果三个最近公共祖先重合，讨论依然成立。设深度数组为$d$，则总距离就等于$$d[a]+d[b]+d[c]-d[x]-2d[y]$$代码如下：

#include <iostream>
#include <cmath>
#include <cstring>
#include <queue>
#include <array>
using namespace std;

const int N = 5e5 + 10, M = N * 2;
int n, m;
int h[N], e[M], ne[M], idx;
int dep[N], f[N][30];
bool vis[N];

#define add(a, b) e[idx] = b, ne[idx] = h[a], h[a] = idx++

void bfs() {
  memset(dep, -1, sizeof dep);
  dep[1] = 0;
  queue<int> q;
  q.push(1);
  vis[1] = true;
  while (q.size()) {
    auto u = q.front(); q.pop();
    for (int i = h[u]; ~i; i = ne[i]) {
      int v = e[i];
      if (vis[v]) continue;
      vis[v] = true;
      dep[v] = dep[u] + 1;
      q.push(v);
      f[v][0] = u;
      for (int k = 1; 1 << k <= dep[v]; k++) f[v][k] = f[f[v][k - 1]][k - 1];
    }
  }
}

int lca(int u, int v) {
  if (dep[u] < dep[v]) swap(u, v);

  for (int k = 0, diff = dep[u] - dep[v]; (1 << k) <= diff; k++)
    if (diff >> k & 1) u = f[u][k];

  if (u == v) return u;

  for (int k = log2(dep[u]) + 1; k >= 0; k--)
    if (f[u][k] != f[v][k]) u = f[u][k], v = f[v][k];

  return f[u][0];
}

int main() {
  memset(h, -1, sizeof h);
  scanf("%d%d", &n, &m);
  for (int i = 1; i < n; i++) {
    int a, b;
    scanf("%d%d", &a, &b);
    add(a, b), add(b, a);
  }

  bfs();

  while (m--) {
    int a, b, c;
    scanf("%d%d%d", &a, &b, &c);
    array<int, 3> s = {lca(a, b), lca(b, c), lca(a, c)};
    if (s[0] == s[1]) swap(s[1], s[2]);
    if (dep[s[0]] < dep[s[1]]) swap(s[0], s[1]);

    printf("%d %d\n", s[0], dep[a] + dep[b] + dep[c] - dep[s[0]] - 2 * dep[s[1]]);
  }
}

预处理时间复杂度$O(N)$，每次询问$O(\log N)$，空间$O(N\log N)$。