[省选] [树链剖分] [BZOJ2243] [SDOI2011] 染色

题目传送门

原本做统计做到吐血，做道树剖醒醒脑，结果写半下午+调半下午……

树剖没的说，那怎么维护颜色呢？

考虑线段树答案的合并，记录三个值：cnt，lc，rc表示这一段的答案，这一段左端点的颜色，这一段右端点的颜色。

合并答案时，如果左子树的右端点和右子树的左端点颜色一致，表示可以中间可以合并成一段，将左右的cnt相加后减一即可。

在链上修改没什么注意的，链上查询时需要记录两边轻链链顶的颜色lc，rc（与之前的那个意义差不多），如果下一次查询的rc和lc一致，答案减一。

查询重链时，注意lc，rc的位置……还要判断相等与否减答案……

感谢zb的数据……

敲个数学作业自动机都得30+行了我是不是要完了…….

敲个树剖都快200行了我是不是要完了……

Code