[2019DecPlat] Bessie's Snow Cow

传送门

官方题解

解答

思路

为处理有关子树的询问,考虑dfs序+区间数据结构,原题可以抽象成以下两个操作:

  1. \([l_x,r_x]\)区间加颜色\(c\)
  2. \([l_x,r_x]\)区间查询颜色数之和。

为了解决1,只能将每种颜色分开处理……对每个颜色开一个称为\(D_1\)的数据结构,存储这个颜色有哪些区间被染色(动态开点线段树!

为了快速处理2,显然只能把所有颜色的结果放在一个区间数据结构\(D_2\)上……但是操作1应该怎么影响\(D_2\)呢?

考察在\([l_x,r_x]\)之间,尚未被染色的区间,将其在\(D_2\)区间加1即可!

复杂度分析

我们注意到,进行操作时,会影响一些区间,但是区间的个数并没有明确的上界。这是不是意味着复杂度会爆炸呢?

当然不是,考虑你一次操作最多增加一个区间,但是每次修改都会在\(D_1\)中删去这个区间,操作次数在均摊意义上是\(O(n)\)的。

\(D\)的选择

综上,\(D_1\)可用set<pair<int,int> >维护,\(D_2\)用线段树或树状数组即可。

posted @ 2021-05-20 20:13  frank3215  阅读(60)  评论(0编辑  收藏  举报