题目链接
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>
<