2021 CCPC 网络赛重赛 Jumping Monkey（并查集重构树，dfs）

最新推荐文章于 2021-10-19 20:47:54 发布

W⁡angduoyu

最新推荐文章于 2021-10-19 20:47:54 发布

阅读量466

点赞数 3

分类专栏：学习总结文章标签：数据结构算法

本文链接：https://blog.youkuaiyun.com/ecjtu2020/article/details/120698074

版权

树结构最远可达节点猴子跳跃问题深度优先搜索连通块

关键词由优快云通过智能技术生成

学习总结专栏收录该内容

39 篇文章

订阅专栏

题目链接

题意

给出一个 $n$ 个结点的树，每个节点有一个权值 $a_i$ 。现在有一个猴子，他可以从 $i$ 结点跳到 $j$ 结点的要求是：在 $i$ -> $j$ 的最短路径上 $j$ 的权值是最大的。现在询问 $\subseteq (1, n)$ 时，猴子能跳到的最多结点数量。

Sol

Code

#include<bits/stdc++.h>
#define int long long
#define endl '\n' 
using namespace std;
typedef pair<int, int> PII;
const int N = 1e5 + 10;
vector<int> G[N]; // 原树
vector<int> NG[N]; // 新树

int depth[N]; // 答案
int p[N];
int find(int x) {
    if(x != p[x]) p[x] = find(p[x]);
    return p[x];
}

void dfs(int u)
{
    for(auto &j : NG[u])
    {
        depth[j] = depth[u] + 1;
        dfs(j);
    }
}

signed main(){
#ifdef ONLINE_JUDGE
#else
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    int T; scanf("%lld", &T);
    while(T -- )
    {
        int n; scanf("%lld", &n);
        for(int i = 0; i <= n; ++i)
        {
            NG[i].clear(); G[i].clear();
            p[i] = i; depth[i] = 0;
        }
        PII a[n + 1];
        vector<bool> vis(n + 1, false);
        for(int i = 1; i < n; ++i)
        {
            int x, y;scanf("%lld%lld", &x, &y);
            G[x].push_back(y); 
            G[y].push_back(x);
        }
        for(int i = 1; i <= n; ++i)
        {
            int x; scanf("%lld", &x);
            a[i] = {x, i};
        }
        sort(a + 1, a + 1 + n);
        vis[a[1].second] = true;
        for(int i = 2; i <= n; ++i)
        {
            int u = a[i].second;
            for(auto &j : G[u])
            {
                if(!vis[j]) continue; // 连接相关边
                int v = find(j);  // 连向该联通块的跟结点
                p[v] = u;
                NG[u].push_back(v);
            }
            vis[u] = true;
        }
        depth[a[n].second] = 1;
        dfs(a[n].second);
        for(int i = 1; i <= n; ++i) cout << depth[i] << endl;
    }
    return 0;
}