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

题目传送门
原本做统计做到吐血，做道树剖醒醒脑，结果写半下午+调半下午…
树剖没的说，那怎么维护颜色呢？
考虑线段树答案的合并，记录三个值：$cnt，lc，rc$表示这一段的答案，这一段左端点的颜色，这一段右端点的颜色。
合并答案时，如果左子树的右端点和右子树的左端点颜色一致，表示可以中间可以合并成一段，将左右的$cnt$相加后减一即可。
在链上修改没什么注意的，链上查询时需要记录两边轻链链顶的颜色$lc，rc$（与之前的那个意义差不多），如果下一次查询的$rc$和$lc$一致，答案减一。
查询重链时，注意$lc，rc$的位置…还要判断相等与否减答案…
感谢zb的数据…
敲个数学作业自动机都得$30+$行了我是不是要完了…
敲个树剖都快200行了我是不是要完了…
Code