题意很简单。对于前两个操作,一个是树链剖分的经典路径更新,另外一个是线段的单点更新。
难点就在于第三个操作怎么来处理。关于子树的更新,很显然跟dfs序有关,如果观察第二个更新重链的dfs,易发现,其实标号的过程也是一个dfs序的过程,只不过优先走的重儿子的路径。
而关于标号跟子树的关系,可以看下图:(图片来源于网络 侵删)
易看出 以u为根节点的子树的标号一定连续,且为: t i d [ u ] − − > t i d [ u ] + s i z [ u ] − 1 tid[u] --> tid[u] + siz[u] - 1 tid[u]−−>tid[u]+siz[u]−1
至此 三个操作均可以转化为线段树区间更新来解决了。
这时还剩最后一个问题,线段树应该维护什么?更新应该更新什么?
如果线段树来维护树剖后每一个节点的值,然后区间求和,那么更新时,每一个节点x的值要么是0(操作2),要么变成一个固定的值val[x](初始的每个节点的值),这样不具备懒操作的区间标记,因为虽然是区间更新,但区间的每一个点增加或减少的值是独立而不相同的。
这时可以利用每个节点x的初始值val[x]不变的性质,我们单独用一个变量v来维护该区间所有节点的val值的和。然后线段树来维护每一个区间对答案的贡献。
比如:若有操作1的区间更新 [ 6 , 10 ] [6,10] [6,10],我们已经知道 [ 6 , 10 ] [6,10] [6,10] 的每一个节点对答案均有贡献,那么直接更新区间和为该区间的v,然后标记一下该区间,同时就没有必要向下继续更新了,这样就可以使用懒操作优化复杂度了。
代码:
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
#define lson rt<<1
#define rson rt<<1|1
const int A = 1e5 + 10;
class Gra{
public:
int v,next;
}G[A<<2];
class Seg_Tree{
public:
int l,r,sum,v,mark;
}Tree[A<<2];
int val[A],head[A],fa[A],son[