树链剖分(模板) 洛谷P3384
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm> #define lson rt<<1, l, mid
#define rson rt<<1|1, mid+1, r
using namespace std;
const int maxn = ;
int P;
typedef long long ll; struct Edge {
int to, next;
}edges[maxn*];
int head[maxn], tot;
int n;
ll w[maxn];
int fa[maxn], son[maxn], siz[maxn], dep[maxn];
int top[maxn], id[maxn], rk[maxn], cnt;
// son[u]: u的重儿子
// top[u]: u所在链的顶端节点
//
// id[u]: 树链剖分后节点的新编号
// rk[u]: dfs编号对应的节点 rk[id[u]] = u
// cnt : dfs序 void add(int u, int v) {
edges[++tot].to = v;
edges[tot].next = head[u];
head[u] = tot;
} // 求fa, siz, dep, son
void dfs1(int u) {
siz[u] = ;
for(int i=head[u];i;i=edges[i].next) {
int v = edges[i].to;
if(v==fa[u]) continue; fa[v] = u;
dep[v] = dep[u] + ;
dfs1(v);
siz[u] += siz[v]; if(siz[v]>siz[son[u]])
son[u] = v;
}
} // 连接重链
void dfs2(int u, int topf) {
id[u] = ++cnt;
rk[cnt] = u; top[u] = topf;
if(!son[u])
return; dfs2(son[u], topf); // 重链先dfs,保证重链上各节点dfs序连续 for(int i=head[u];i;i=edges[i].next) {
int v = edges[i].to;
if(v!=fa[u] && v!=son[u]) // 不是重儿子,轻链
dfs2(v, v);
}
} /*
操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式: 2 x y 表示求树从x到y结点最短路径上所有节点的值之和 操作3: 格式: 3 x z 表示将以x为根节点的子树内所有节点值都加上z 操作4: 格式: 4 x 表示求以x为根节点的子树内所有节点值之和
*/ ll sum[maxn*], lazy[maxn*]; void build(int rt, int l, int r) {
if(l!=r) {
int mid = (l+r)/;
build(lson);
build(rson);
sum[rt] = (sum[rt<<] + sum[rt<<|]) % P;
} else {
sum[rt] = w[rk[l]];
lazy[rt] = ;
}
}
void pushDown(int rt, int len) {
if(lazy[rt]) {
(sum[rt<<] += (len-(len>>)) * lazy[rt] % P) %= P;
(sum[rt<<|] += (len>>) * lazy[rt] % P) %= P; (lazy[rt<<] += lazy[rt]) %= P;
(lazy[rt<<|] += lazy[rt]) %= P;
lazy[rt] = ;
}
} void update(int rt, int l, int r, int L, int R, ll add) {
if(L<=l && R>=r) {
(lazy[rt] += add) %= P;
(sum[rt] += (r-l+)*add) %= P;
return;
} int mid = (l+r)/;
pushDown(rt, r-l+); if(L<=mid)
update(lson, L, R, add);
if(R>mid)
update(rson, L, R, add); sum[rt] = (sum[rt<<] + sum[rt<<|]) % P;
} ll query(int rt, int l, int r, int L, int R) {
if(L<=l && R>=r) return sum[rt]; ll res = ;
int mid = (l+r)/;
pushDown(rt, r-l+); if(L<=mid) res += query(lson, L, R) % P;
if(R>mid) res += query(rson, L, R) % P;
return res % P;
} // 操作1
void updatePath(int x, int y, ll z) {
while(top[x]!=top[y]) {
if(dep[top[x]]<dep[top[y]]) swap(x, y);
update(, , n, id[top[x]], id[x], z);
x = fa[top[x]];
}
if(dep[x]>dep[y]) swap(x, y);
update(, , n, id[x], id[y], z);
} // 操作3
void updateSon(int x, ll z) {
update(, , n, id[x], id[x]+siz[x]-, z);
} // 操作2
ll queryPath(int x, int y) {
ll res = ;
while(top[x]!=top[y]) {
if(dep[top[x]]<dep[top[y]]) swap(x, y);
res = (res + query(, , n, id[top[x]], id[x])) % P;
x = fa[top[x]];
}
if(dep[x]>dep[y]) swap(x, y);
res = (res + query(, , n, id[x], id[y])) % P;
return res;
} // 操作4
ll qeurySon(int x) {
return query(, , n, id[x], id[x]+siz[x]-);
} int main() {
int M, R;
cin>>n>>M>>R>>P;
for(int i=;i<=n;i++) {
scanf("%lld", &w[i]);
}
for(int i=;i<n-;i++) {
int u, v;
scanf("%d %d", &u, &v);
add(u, v);
add(v, u);
} dfs1(R);
dfs2(R, R); build(, , n); while(M--) {
int op, x, y;
ll z;
scanf("%d", &op);
if(op==) {
scanf("%d %d %lld", &x, &y, &z);
updatePath(x, y, z);
} else if(op==) {
scanf("%d %d", &x, &y);
printf("%lld\n", queryPath(x, y));
} else if(op==) {
scanf("%d %lld", &x, &z);
updateSon(x, z);
} else {
scanf("%d", &x);
printf("%lld\n", qeurySon(x));
} }
return ;
}
树链剖分(模板) 洛谷P3384的更多相关文章
- 树链剖分模板(洛谷P3384)
洛谷P3384 #include <bits/stdc++.h> #define DBG(x) cerr << #x << " = " < ...
- 【树链剖分】洛谷P3384树剖模板
题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式 ...
- 树链剖分( 洛谷P3384 )
我们有时候遇到这样一类题目,让我们维护树上路径的某些信息,这个时候发现我们无法用线段树或者树状数组来维护这些信息,那么我们就有着一种新的数据结构,树剖:将一棵树划分成若干条链,用数据结构去维护每条链, ...
- 【树链剖分】洛谷P3379 树链剖分求LCA
题目描述 如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先. 输入输出格式 输入格式: 第一行包含三个正整数N.M.S,分别表示树的结点个数.询问的个数和树根结点的序号. 接下来N-1行每 ...
- 树链剖分【洛谷P4114】 Qtree1
P4114 Qtree1 题目描述 给定一棵n个节点的树,有两个操作: CHANGE i ti 把第i条边的边权变成ti QUERY a b 输出从a到b的路径中最大的边权,当a=b的时候,输出0 码 ...
- 树链剖分【洛谷P2590】 [ZJOI2008]树的统计
P2590 [ZJOI2008]树的统计 题目描述 一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w. 我们将以下面的形式来要求你对这棵树完成一些操作: I. CHANGE u t : 把 ...
- 树链剖分【洛谷P1505】 [国家集训队]旅游
P1505 [国家集训队]旅游 题目描述 Ray 乐忠于旅游,这次他来到了T 城.T 城是一个水上城市,一共有 N 个景点,有些景点之间会用一座桥连接.为了方便游客到达每个景点但又为了节约成本,T 城 ...
- 树链剖分【洛谷P3833】 [SHOI2012]魔法树
P3833 [SHOI2012]魔法树 题目描述 Harry Potter 新学了一种魔法:可以让改变树上的果子个数.满心欢喜的他找到了一个巨大的果树,来试验他的新法术. 这棵果树共有N个节点,其中节 ...
- BZOJ 2243 染色 | 树链剖分模板题进阶版
BZOJ 2243 染色 | 树链剖分模板题进阶版 这道题呢~就是个带区间修改的树链剖分~ 如何区间修改?跟树链剖分的区间询问一个道理,再加上线段树的区间修改就好了. 这道题要注意的是,无论是线段树上 ...
- 算法复习——树链剖分模板(bzoj1036)
题目: 题目背景 ZJOI2008 DAY1 T4 题目描述 一棵树上有 n 个节点,编号分别为 1 到 n ,每个节点都有一个权值 w .我们将以下面的形式来要求你对这棵树完成一些操作:I.CHAN ...
随机推荐
- PHP 类中静态方法调用非静态方法
静态方法调用非静态方法: 在类中静态方法中,需要实例化对象,然后再调用类中的方法 非静态方法调用静态方法: 可以self 或者 类名加::的形式调用 如下面的案例: <?php class A{ ...
- 【JZOJ6409】困难的图论
description 给定由 n 个点 m 条边组成的无向连通图,保证没有重边和自环. 你需要找出所有边,满足这些边恰好存在于一个简单环中.一个环被称为简单环,当且仅当它包含的所有点都只在这个环中被 ...
- HIVE的数据类型
- selenium 显示等待、隐式等待、强制等待
如今大部分web程序使用Ajax技术,当浏览器加载页面时,页面元素可能不是同时加载完成,如果因为加载某个元素超时导致ElementNotVisibleException的情况出现,自动化脚本的稳定性就 ...
- thinkphp 缓存驱动
缓存驱动默认位于Think\Cache\Driver命名空间下面,目前已经提供了包括APC.Db.Memcache.Shmop.Sqlite.Redis.Eaccelerator和Xcache缓存方式 ...
- 利用html2canvas截图,得到base64上传ajax
<script type="text/javascript" src="js/html2canvas.js"></script> //布 ...
- NX二次开发-UFUN工程图导入视图UF_DRAW_import_view
NX9+VS2012 #include <uf.h> #include <uf_draw.h> #include <uf_obj.h> #include <u ...
- 源码安装zabbix-oracle
源码安装zabbix_agent4.0.3 1.源码包下载地址:https://www.zabbix.com/download_sources 2.下载完后上传在任意目录用root用户创建以下脚本 ...
- GO语言range的用法 (2013-08-09 14:08:26)
range是go语言系统定义的一个函数. 函数的含义是在一个数组中遍历每一个值,返回该值的下标值和此处的实际值. 假如说a[0]=10,则遍历到a[0]的时候返回值为0,10两个值. 下面是一个例子: ...
- 修改linux默认语言
linux修改默认语言 编辑/etc/sysconfig/i18n这个文件(不存在就新建一个),原内容如下: LANG="en_US.UTF-8" SYSFONT=" ...