题意 :
- n点的树,每个点的点权distinct,求分别从每个点出发能到达的点数(被到达的点必须是路径上点权最大的点)
思路 :
- 考虑点权最大的那个结点(假设为u),显然不管从哪个结点开始跳一次,都可以跳到u,且不能越过u,也就是说到达u后不能跳到其它点,于是u一定是最优方案中最后一个到达的点。我们将结点u以及它相连的边从树上去掉,对剩下的每个连通块递归进行上述过程即可获得答案。
- 但要实现上述递归过程并不容易,转而反向思考这个过程,按点权从小到大枚举结点u,并将u作为所有与u相连的(已经枚举过的)连通块的根(用并查集维护,只连一个点可以将复杂度从O(n)降到O(logn))从而建立一颗新树,每个结点的深度(根深度为1)即是答案
#include <iostream>
#include <cstring>
#include <map>
#include <vector>
using namespace std;
const int N = 1e5 + 10;
vector<int> G[N], G2[N];
int fa[N];
bool st[N];
int c[N], a[N];
map<int, int> dc;
void init(int n)
{
for (int i = 1; i <= n; i ++ )
fa[i] = i;
}
int find(int x)
{
if (fa[x] != x) fa[x] = find(fa[x]);
return fa[x];
}
void merge(int a, int b)
{
a = find(a), b = find(b);
if (a != b) fa[a] = b;
}
// 求点的深度
void dfs(int u)
{
for (auto v : G2[u])
{
a[v] = a[u] + 1;
dfs(v);
}
}
int main()
{
ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
int T;
cin >> T;
while (T -- )
{
int n;
cin >> n;
// 并查集初始化
init(n);
// 多组数据
for (int i = 1; i <= n; i ++ ) G[i].clear(), G2[i].clear();
memset(st, 0, sizeof st);
dc.clear();
// 树,无向图,注意是n - 1条边!!!
for (int i = 0; i < n - 1; i ++ )
{
int u, v;
cin >> u >> v;
G[u].push_back(v);
G[v].push_back(u);
}
// map 点权-点编号
for (int i = 1; i <= n; i ++ )
{
cin >> c[i];
dc[c[i]] = i;
}
st[dc.begin() -> second] = true; // 点权值最小的点为根,第一个访问
for (auto it = ++ dc.begin(); it != dc.end(); it ++ )
{
auto u = it -> second;
for (auto v : G[u]) // 原图中相连的点
{
if (!st[v]) continue; // 已经枚举过的连通块
int cc = find(v);
merge(v, u); // 将已经枚举过的连通块的根接在u下
G2[u].push_back(cc);
}
st[u] = true; // 已经访问过u
}
int x = ( -- dc.end()) -> second;
a[x] = 1;
dfs(x);
for (int i = 1; i <= n; i ++ ) cout << a[i] << endl;
}
return 0;
}