树链剖分(模板) 洛谷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 ...
随机推荐
- 报错 DOMDocument not found
php -m 查看有没有dom扩展 没有安装扩展 yum install php-dom php 常用扩展有 yum install php-solr php-opcache php-seasLog ...
- 19. 接口(创建interface 继承implements)
1.语法: interface 接口名{ 属性 抽象方法 } 2.实例: 3.注意: 1)类实现接口可以通过implements实现,实现接口的时候必须把接口中的所有方法实现,一个类可以实现多个接口. ...
- wpf 绑定除数据上下文外的属性
例如: listview 绑定了一个windows.datacontext.一个集合,那么其中一个item想绑定windows.datacontext.A属性怎么办: 通过查找祖先的方法:
- element中的tree组件实现菜单分配
返回的菜单数据 tree组件的使用 <el-tree ref="menuList" // :data="menuList" // 展示数据 :props= ...
- 暴力模拟——cf988E
很简单的题,就是模拟一下 #include<bits/stdc++.h> using namespace std; #define ll long long ll n,a[],len; i ...
- 动态隐藏显示窗口的标题栏(同时保持窗口的sizeable性能
今天考虑作界面的时候,想去掉窗体的标题栏,但设置 Form.BorderStyle := bsNone; 会导致窗体不可再 Sizeable (通过鼠标操作改变窗体大小),仔细翻看了相关Help也没找 ...
- idea使用问题
1. 问题: 突发断电导致idea的play项目错误,无法识别build.sbt,build.sbt文件报错,Cannot resolve symbol 解决方案: For anyone having ...
- iOS ARC使用总结
在iOS ARC模式下,编译器会自动插入release 有些时候程序出现 message sent to deallocated object的时候,你不知道什么原因. 一种原因是因为你在ARC下使用 ...
- D 语言学习感受
胃抽动!特想吐!不过,还是很佩服写这种语言的人,不是大拿,怎么可能开发一门语言!
- mac 安装配置使用 mongoldb
mac 安装配置使用 mongoldb 安装和配置 brew install mongos brew install mongo # 密码就是用户的密码 # 配置数据文件 //如果不配置会出现错误62 ...