洛谷U41492 树上数颜色树上启发式合并

最新推荐文章于 2024-04-21 08:13:53 发布

moomhxy

最新推荐文章于 2024-04-21 08:13:53 发布

阅读量362

点赞数

分类专栏： DSU on tree 洛谷

本文链接：https://blog.csdn.net/qq_41785863/article/details/106910092

版权

该博客详细解析了洛谷比赛中编号为U41492的题目，内容围绕如何解决在一棵根为1的树中，针对每个子树查询颜色种类数的问题。作者提出了运用树上启发式合并的策略，解释了在处理过程中，首次遇到节点时计算其颜色贡献，以及处理轻链时消除贡献的关键步骤。

摘要由CSDN通过智能技术生成

题目链接：https://www.luogu.com.cn/problem/U41492
题意：给一棵根为1的树，每次询问子树颜色种类数
思路：树上启发式合并，每次计算时如果是第一次出现就计算贡献，轻链的话要消除贡献

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define fi first
#define se second
#define ls rt << 1
#define rs rt << 1|1
#define po pop_back
#define pb push_back
#define mk make_pair
#define lson l, mid, ls
#define rson mid + 1, r, rs
#define pll pair<ll, ll>
#define pii pair<int, int>
#define ull unsigned long long
#define pdd pair<double, double>
const int mod = 1e9 + 9;
const int maxn = 2e5 + 10;
const int inf = 0x3f3f3f3f;
const ll linf = 0x3f3f3f3f3f3f3f3f;
int c[maxn], num[maxn], sz[maxn], son[maxn], sum[maxn], cnt[maxn], ans[maxn], Max, Son;
vector<int> e[maxn];

void dfs(int u, int fa) //统计重儿子信息
{
    sz[u] = 1;
    //sum[u] = num[u];
    for(auto v : e[u])
    {
        i