题意概述:给出一棵树,初始每个结点有不同的颜色。现在支持三种操作:1.把某个结点的颜色改成一个之前都没有出现过的颜色,并将这个点到当前树根路径上的所有点全部改成这个颜色;2.改变当前的树根到另外一个点,并对原来的树根进行一次操作1;3.把询问当前形态的树中对一个点的子树中所有点进行操作1的平均代价(操作1代价的计算方式:这个点到当前树根路径上的不同颜色数量)。
从原题干来分析,相同的颜色一定连续出现在一条链上。再仔细想想,这样的操作正是LCT中的access操作!而每一次操作的代价就是access的时候经历的虚边的数量。如果是单点询问的话这就是一道LCT的裸题,但是问题在于题目要求的是子树询问。对子树信息的维护很容易想到用DFS序配合线段树来搞定。(于是就有两种流派:线段树单独在外面维护子树信息和就在LCT中维护子树信息两种,我选择线段树,常数小啊!如果要Link和Cut的话大不了我再来个LCT维护一下DFS序。。。)
分析没有换根操作的情况,在access操作中,令函数调用的时候的对象为s,当前所在结点为x,上一次所在的splay的根结点为y。y所在的splay就是s->y所在splay权值(真实树中深度)最小的点的一条链。我们找到链顶,可以发现这个链顶的子树中所有及结点在这一次access跳跃之后答案都要-1(每个点到根上的路径都少了一种颜色),x在当前树中位于splay维护的链上的儿子的子树中所有结点的答案+1。一路维护上去所有被影响的点的答案都被更新了。
换根的情况?可以发现在没有换根操作的时候可以保证对每个点的子树进行操作的时候一定是对应dfs序中一段连续的序列,但是换根之后就可以出现当前某个点子树在以1为根的dfs序中序列不连续,但是实际上也可以发现就是被分成了两端,x的子树对应的dfs序就是把x当前的父亲从dfs序中挖掉剩下的部分。对于是不是要挖点的判断借助DFS序就可以做到了。
随便分析几下思路就有了,也不是很难码,关键是细节,各种情况考虑全(为了节省您的时间请一定把一切想清楚了再开始码)。。。。。。这里只说一点:注意对LCT每个点的儿子的正确维护!
1 #include 2 #include 3 #include 4 #include 5 #include 6 #include 7 #include 8 #include 9 #include