题目描述
有一棵点数为 N 的树,以点 1 为根,且树点有边权。然后有 M 个操作,分为三种:
-
操作 1 :把某个节点 x 的点权增加 a 。
-
操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a 。
-
操作 3 :询问某个节点 x 到根的路径中所有点的点权和。
输入格式
第一行包含两个整数 N, M 。表示点数和操作数。 接下来一行 N 个整数,表示树中节点的初始权值。 接下来 N-1 行每行两个正整数 from, to , 表示该树中存在一条边 (from, to) 。 再接下来 M 行,每行分别表示一次操作。其中第一个数表示该操作的种类( 1-3 ) ,之后接这个操作的参数( x 或者 x a ) 。
输出格式
对于每个询问操作,输出该询问的答案。答案之间用换行隔开。
思路
如果你会树链剖分的话,这个题就是个树链剖分的裸题,但是还有更好的解法,帮助锻炼思维。
线段树+欧拉序。
对每个点维护两个东西,一个入度的dfs序,一个出度的dfs序。
这样,我们将一个点拆成两个,对第一个点加权,第二个点减权,这样,对于点x,答案就是1到in[x]的区间和。
比如下面的图
我们按照dfs序列遍历就会得到以下的序列