bzoj 1483 [HNOI2009]梦幻布丁 线段树合并

题目链接

https://www.lydsy.com/JudgeOnline/problem.php?id=1483

做法

网上好多的做法都是链表合并，我感觉有点难写，这里提供一种不一样的思路，线段树合并，复杂度是一样的，都是一个log，但是代码复杂度应该会比较简单。
我们对每个颜色建一棵线段树，线段树每个节点记录三个信息，size，ll，rr，分别表示这个颜色这段区间中的段数，左边是否顶到，右边是否顶到。
update的时候，size[u]=size[lson[u]+size[rson]-(rr[lson[u]&&ll[rson[u])，应该可以理解吧。
那么一开始的答案就是所有size[root[i]]的和。
对于修改操作，我们先ans-=size[root[x]]+size[root[y]]，然后我们把x的这棵线段树合并到y上面，因为底层的节点不会重复，所以我们先递归到底层，然后不断update上来就行了，这里的合并使用启发式合并，保证复杂度的正确性。最后ans在加上size[root[y]]就行了，然后记得把root[x]设为0。
注意判一下x==y的情况就行了。

代码

#include<cstdio>
<