模板 - 图论 - 树链剖分
今天来啃一下这个树剖吧。
模板题是要求这四个问题:
- 将树从x到y结点最短路径上所有节点的值都加上z
- 求树从x到y结点最短路径上所有节点的值之和
- 将以x为根节点的子树内所有节点值都加上z
- 求以x为根节点的子树内所有节点值之和
http://www.cnblogs.com/zwfymqz/p/8094500.html
这里的边要开两倍,线段树当然要开四倍。
第一次模板:
#include<bits/stdc++.h> using namespace std; #define ll long long/* 树链剖分 begin*/#define MAXN 100100 #define ls k<<1 #define rs k<<1|1struct Edge{int u,v,nxt; }edge[MAXN*2];int head[MAXN]; int num=1;void addedge(int u,int v){edge[num].u=u;edge[num].v=v;edge[num].nxt=head[u];head[u]=num++; }int deep[MAXN];//节点的深度 int fa[MAXN];//节点的父亲 int son[MAXN];//节点的重儿子 int tot[MAXN];//节点子树的大小 int top[MAXN]; int idx[MAXN];int cnt=0; int a[MAXN]; int b[MAXN]; int MOD;int dfs1(int now, int f, int dep) {deep[now] = dep;fa[now] = f;tot[now] = 1;int maxson = -1;for (int i = head[now]; i != -1; i = edge[i].nxt) {if (edge[i].v == f) continue;tot[now] += dfs1(edge[i].v, now, dep + 1);if (tot[edge[i].v] > maxson) maxson = tot[edge[i].v], son[now] = edge[i].v;}return tot[now]; }void dfs2(int now, int topf) {idx[now] = ++cnt;a[cnt] = b[now];top[now] = topf;if (!son[now]) return ;dfs2(son[now], topf);for (int i = head[now]; i != -1; i = edge[i].nxt)if (!idx[edge[i].v])dfs2(edge[i].v, edge[i].v); }struct Tree {int l, r, w, siz, f; } T[MAXN*4];void update(int k) { //更新T[k].w = (T[ls].w + T[rs].w + MOD) % MOD; }void Build(int k, int l, int r) {T[k].l = l; T[k].r = r; T[k].siz = r - l + 1;if (l == r) {T[k].w = a[l];return ;}int mid = (l + r) >> 1;Build(ls, l, mid);Build(rs, mid + 1, r);update(k); }void pushdown(int k) { //下传标记if (!T[k].f) return ;T[ls].w = (T[ls].w + T[ls].siz * T[k].f) % MOD;T[rs].w = (T[rs].w + T[rs].siz * T[k].f) % MOD;T[ls].f = (T[ls].f + T[k].f) % MOD;T[rs].f = (T[rs].f + T[k].f) % MOD;T[k].f = 0; }void IntervalAdd(int k, int l, int r, int val) { //区间加if (l <= T[k].l && T[k].r <= r) {T[k].w += T[k].siz * val;T[k].f += val;return ;}pushdown(k);int mid = (T[k].l + T[k].r) >> 1;if (l <= mid) IntervalAdd(ls, l, r, val);if (r > mid) IntervalAdd(rs, l, r, val);update(k); }int IntervalSum(int k, int l, int r) { //区间求和int ans = 0;if (l <= T[k].l && T[k].r <= r)return T[k].w;pushdown(k);int mid = (T[k].l + T[k].r) >> 1;if (l <= mid) ans = (ans + IntervalSum(ls, l, r)) % MOD;if (r > mid) ans = (ans + IntervalSum(rs, l, r)) % MOD;return ans; }int TreeSum(int x, int y) { //求x与y路径上的和int ans = 0;while (top[x] != top[y]) {if (deep[top[x]] < deep[top[y]]) swap(x, y);ans = (ans + IntervalSum(1, idx[ top[x] ], idx[x])) % MOD;x = fa[ top[x] ];}if (deep[x] > deep[y]) swap(x, y);ans = (ans + IntervalSum(1, idx[x], idx[y])) % MOD;return ans; }void TreeAdd(int x, int y, int val) { //对于x,y路径上的点加val的权值while (top[x] != top[y]) {if (deep[top[x]] < deep[top[y]]) swap(x, y);IntervalAdd(1, idx[ top[x] ], idx[x], val);x = fa[ top[x] ];}if (deep[x] > deep[y]) swap(x, y);IntervalAdd(1, idx[x], idx[y], val); }//求x所在子树的和 //IntervalSum(1,idx[x],idx[x]+tot[x]-1); //对x所在子树的点加val的权值 //IntervalAdd(1,idx[x],idx[x]+tot[x]-1,val%MOD);/* 树链剖分 end */int main(){memset(head,-1,sizeof(head));int n,m,r,p;cin>>n>>m>>r>>p;MOD=p;for(int i=1;i<=n;i++){cin>>b[i];}for(int i=1;i<=n-1;i++){int u,v;cin>>u>>v;addedge(u,v);addedge(v,u);}dfs1(r,0,1);dfs2(r,r);Build(1,1,n);while(m--){int opt,x,y,z;cin>>opt;switch(opt){case 1:cin>>x>>y>>z;TreeAdd(x,y,z%p);break;case 2:cin>>x>>y;cout<<TreeSum(x,y)<<endl;break;case 3:cin>>x>>z;IntervalAdd(1,idx[x],idx[x]+tot[x]-1,z%p);break;case 4:cin>>x;cout<<IntervalSum(1,idx[x],idx[x]+tot[x]-1)<<endl;}} }
View Code
转载于:https://www.cnblogs.com/Yinku/p/10652569.html
模板 - 图论 - 树链剖分相关推荐
- 洛谷 P3384 【模板】树链剖分-树链剖分(点权)(路径节点更新、路径求和、子树节点更新、子树求和)模板-备注结合一下以前写的题目,懒得写很详细的注释...
P3384 [模板]树链剖分 题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节 ...
- 专题·树链剖分【including 洛谷·【模板】树链剖分
初见安~~~终于学会了树剖~~~ [兴奋]当初机房的大佬在学树剖的时候我反复强调过:"学树剖没有前途的!!!" 恩.真香. 一.重链与重儿子 所谓树剖--树链剖分,就是赋予一个链的 ...
- 【模板】树链剖分 P3384
题目链接 //部分转自:https://www.luogu.org/problemnew/solution/P3384 初学树链剖分,感觉这个模板题还是容易理解的,但是实在是码量很大的. 知识点: 重 ...
- 洛谷3384:【模板】树链剖分——题解
https://www.luogu.org/problemnew/show/P3384 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 ...
- 洛谷 P3384 【模板】树链剖分
题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式 ...
- P3384 【模板】树链剖分
chuansongmen 题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的 ...
- 树链剖分 讲解+模板+习题
今天我们来讲一下树链剖分 树链剖分是什么? 树链剖分是一种用来维护树上路径信息的在线方法,可以处理在线. 通常通过一种方法,将一棵树剖分成若干条链,然后通过数据结构(线段树,BIT等)去维护. 我们通 ...
- 归纳(四):树链剖分
剖分的意义 能用线段树搞想要的信息. (其实可以只用来求LCA) 需要的东西 七个数组,在两次dfs中处理出来. dfs1: dep[]:深度 fa[]:父亲 son[]:重儿子 siz[]:子树大小 ...
- 【暖*墟】#树链剖分# 树链剖分学习与练习
树链剖分 树链剖分是一种优化,将树上最常经过的几条链划为重点,用线段树来优化区间修改和查询. 并且因为在一棵子树中dfs序是连续的,并且在任意一条重链上,dfs序也是连续的, 可以认为轻链是单点修改, ...
最新文章
- Frighting的日常:第7天
- DL之GD:利用LogisticGD算法(梯度下降)依次基于一次函数和二次函数分布的数据集实现二分类预测(超平面可视化)
- lnmp里面php测试脚本,军哥LNMP V1.4测试版一键脚本安装以及功能上的升级体验
- WCF存储图片到指定文件夹下
- jQuery实用Ajax loading旋转指示器插件
- python爬取今日热榜数据到txt文件
- Java基础知识总结二(2)
- 【语音识别】基于matlab GUI BP神经网络0到10数字语音识别【含Matlab源码 672期】
- linux命令iconv_linux命令系列之iconv
- VMWare16下载安装Centos7镜像教程,详细记录
- oracle年龄计算函数
- 人员疏散模型(pso元胞自动机)网挑思路
- Android 隐藏程序的图标
- [c++] 什么是平凡类型,标准布局类型,POD类型,聚合体
- win7资源管理器总是崩溃
- leet75:颜色分类
- iOS复习中有关SDWebImage可能知识点总结
- 【复】一次流量分析经历
- 腾讯——云计算从业者认证系列1
- 【5G NR】物理资源