2019 ICPC 南昌站 K Tree 树上启发式合并 + 动态开点线段树

最新推荐文章于 2022-10-26 16:13:48 发布

原创

最新推荐文章于 2022-10-26 16:13:48 发布 · 198 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

传送门

题意

在这里插入图片描述

分析

首先，要求一个点不为另一个点的祖先，这个比较好处理，先确定一个根结点 $u$ ，在他的两颗子树里面找点就可以了
两个点和权值和为定值，可以用离散化+树上启发式合并进行处理
剩下的就是路径长度了，这个不太好处理，因为是求一个范围内的合法点的数量，怎么办呢，我们可以用权值线段树来处理，每一个点权建一颗线段树，表示每一个深度的点出现了多少次，因为范围比较大，所以需要动态开点

代码

#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
#define _CRT_SECURE_NO_WARNINGS
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;
typedef vector<int