当前位置:首页 > 数据结构 > 树链剖分 > 正文
洛谷P3384【模板】树链剖分
2626+

题目大意:一棵树,请实现两点之间的最短路径上的结点加数、求和以及子树上的结点加数、求和操作。

题目描述

如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作:

操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z

操作2: 格式: 2 x y 表示求树从x到y结点最短路径上所有节点的值之和

操作3: 格式: 3 x z 表示将以x为根节点的子树内所有节点值都加上z

操作4: 格式: 4 x 表示求以x为根节点的子树内所有节点值之和

输入输出格式

输入格式:

第一行包含4个正整数N、M、R、P,分别表示树的结点个数、操作个数、根节点序号和取模数(即所有的输出结果均对此取模)。

接下来一行包含N个非负整数,分别依次表示各个节点上初始的数值。

接下来N-1行每行包含两个整数x、y,表示点x和点y之间连有一条边(保证无环且连通)

接下来M行每行包含若干个正整数,每行表示一个操作,格式如下:

操作1: 1 x y z

操作2: 2 x y

操作3: 3 x z

操作4: 4 x

输出格式:

输出包含若干行,分别依次表示每个操作2或操作4所得的结果(对P取模

输入输出样例

输入样例#1:

5 5 2 24
7 3 7 8 0 
1 2
1 5
3 1
4 1
3 4 2
3 2 2
4 5
1 5 1 3
2 1 3
输出样例#1:

2
21

说明

时空限制:1s,128M

数据规模:

对于30%的数据:

对于70%的数据:

对于100%的数据:

解题思路

树链剖分:预处理各个结点的父亲、重儿子、深度、子树结点数量,按住DFS序重新编号、设置重链链头,如果是a到b的路径加数,那么一条一条链区间加;如果是子树加数,子树编号连续,直接区间加;求和方法类似。如果不想分块写求和、加数爬树处理函数,可以写在一起:如果是加,就直接加,如果是求和,就求和到ans,最后返回ans的值,加的时候不输出,求和的时候才输出。注意:线段树中求和、乘区间数量的时候可能会爆int,用long long比较保险,另外多取几次mo可避免出错。

程序实现

About

坚决不Copy代码!

本文标签:,,,,

洛谷P3384【模板】树链剖分:等您坐沙发呢!

发表评论

您必须 [ 登录 ] 才能发表留言!