P4149 [IOI2011]Race (dsu on tree处理路径问题)

tp链接

对于dis(u,v)==k这样的路径问题，
我们可以借用LCA将其转化为dis(1,u) + dis(1,v)-2*dis(1,lca) == k
问题就成了 ，满足上述式子的情况下，求出:
dep(u) + d(v) - 2*dep(lca) 的最小值

我们可以枚举LCA来求解。

枚举LCA就联想到可以枚举每颗子树，根节点作为LCA，于是问题转化成了子树查询。

每颗子树的O(N)暴力解法：维护一个map,map[x]表示dis值为x的深度最小的节点深度。 枚举其孩子子树，先查询并更新答案，然后再维护map(枚举LCA时的经典老活了)。
发现最后一棵子树的信息是可以继承给rt的，所以dsu on tree。

这题用到的小trick:

• 路径距离问题通过LCA转化成子树查询问题，再用dsu on tree
• 枚举LCA的子树查询问题往往需要先查询答案，再维护信息
• 枚举LCA的时候要注意LCA这个点是否会影响答案，不能遗漏

#include<bits/stdc++.h>
using namespace std;
//#pragma GCC optimize(2)
#define ull unsigned long long
#define ll long long
#define pii pair<int, int>
#define pdd pair<double, double>
#define re register
const int maxn = 2e5 + 10;
const ll mod = 998244353;
const ll inf = (ll)4e17+5;
const int INF = 1e9 + 7;
const double pi = acos(-1.0);
//给定带边权的树 求出一条最短的简单路径满足长度为K 
//dis[i]+dis[j]-2dis[LCA]==k  dep[i]+dep[j]-2*dep[LCA] 最小 
//我们枚举LCA 对于每个LCA求出答案取min 
//LCA即树根 所以等价于枚举每颗子树
//维护一个map 表示离根节点dis为i的深度最小值
vector<pii> g[maxn];
int n,k;