P6847 [CEOI2019] Magic Tree

提供一种别样的线段树合并写法。

\mathcal MagicMagic

魔法树

给定一棵 nn 个节点的树，有些节点上有果实，有值 d,wd,w，分别表示果实成熟的日期和价值。

如果恰好在日期当天切掉该节点与根的连通，那么就能得到该价值。求能得到的价值的最大值。

转化一下题意，在树上选择一些点，使得所有被选择的点对 (u,v)(u,v)，若 vv 在以 uu 为根的子树中，那么有 d(v)\leq d(u)d(v)≤d(u)。

考虑最朴素的树形 DP，设 f(u,i)f(u,i) 表示以 uu 为根的子树，所有切断操作都在 \leq i≤i 天内完成的最大价值。

那么对于 uu 的一个儿子 vv，它的贡献有：

1. 不考虑加入 uu 点的价值：f(u,i)=\sum f(v,i)f(u,i)=∑f(v,i)。
2. 考虑加入 uu 点的价值：f(u,j)=w(u)+\sum f(v,d(u))f(u,j)=w(u)+∑f(v,d(u))，其中 j\geq d(u)j≥d(u)。

因为两者是互斥关系，所以需要取个 \maxmax。

这样就得到了 O(nk)O(nk) 的暴力算法，考虑使用数据结构优化它，不难想到线段树合并。

对于转移一可以直接合并，打上个区间加标价即可，对于二相当于是区间 [d(u),k][d(u),k] 对那个值取 \maxmax。

<