今天来啃一下这个树剖吧。

模板题是要求这四个问题:

  • 将树从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

模板 - 图论 - 树链剖分相关推荐

  1. 洛谷 P3384 【模板】树链剖分-树链剖分(点权)(路径节点更新、路径求和、子树节点更新、子树求和)模板-备注结合一下以前写的题目,懒得写很详细的注释...

    P3384 [模板]树链剖分 题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节 ...

  2. 专题·树链剖分【including 洛谷·【模板】树链剖分

    初见安~~~终于学会了树剖~~~ [兴奋]当初机房的大佬在学树剖的时候我反复强调过:"学树剖没有前途的!!!" 恩.真香. 一.重链与重儿子 所谓树剖--树链剖分,就是赋予一个链的 ...

  3. 【模板】树链剖分 P3384

    题目链接 //部分转自:https://www.luogu.org/problemnew/solution/P3384 初学树链剖分,感觉这个模板题还是容易理解的,但是实在是码量很大的. 知识点: 重 ...

  4. 洛谷3384:【模板】树链剖分——题解

    https://www.luogu.org/problemnew/show/P3384 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 ...

  5. 洛谷 P3384 【模板】树链剖分

    题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式 ...

  6. P3384 【模板】树链剖分

    chuansongmen 题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的 ...

  7. 树链剖分 讲解+模板+习题

    今天我们来讲一下树链剖分 树链剖分是什么? 树链剖分是一种用来维护树上路径信息的在线方法,可以处理在线. 通常通过一种方法,将一棵树剖分成若干条链,然后通过数据结构(线段树,BIT等)去维护. 我们通 ...

  8. 归纳(四):树链剖分

    剖分的意义 能用线段树搞想要的信息. (其实可以只用来求LCA) 需要的东西 七个数组,在两次dfs中处理出来. dfs1: dep[]:深度 fa[]:父亲 son[]:重儿子 siz[]:子树大小 ...

  9. 【暖*墟】#树链剖分# 树链剖分学习与练习

    树链剖分 树链剖分是一种优化,将树上最常经过的几条链划为重点,用线段树来优化区间修改和查询. 并且因为在一棵子树中dfs序是连续的,并且在任意一条重链上,dfs序也是连续的, 可以认为轻链是单点修改, ...

最新文章

  1. Frighting的日常:第7天
  2. DL之GD:利用LogisticGD算法(梯度下降)依次基于一次函数和二次函数分布的数据集实现二分类预测(超平面可视化)
  3. lnmp里面php测试脚本,军哥LNMP V1.4测试版一键脚本安装以及功能上的升级体验
  4. WCF存储图片到指定文件夹下
  5. jQuery实用Ajax loading旋转指示器插件
  6. python爬取今日热榜数据到txt文件
  7. Java基础知识总结二(2)
  8. 【语音识别】基于matlab GUI BP神经网络0到10数字语音识别【含Matlab源码 672期】
  9. linux命令iconv_linux命令系列之iconv
  10. VMWare16下载安装Centos7镜像教程,详细记录
  11. oracle年龄计算函数
  12. 人员疏散模型(pso元胞自动机)网挑思路
  13. Android 隐藏程序的图标
  14. [c++] 什么是平凡类型,标准布局类型,POD类型,聚合体
  15. win7资源管理器总是崩溃
  16. leet75:颜色分类
  17. iOS复习中有关SDWebImage可能知识点总结
  18. 【复】一次流量分析经历
  19. 腾讯——云计算从业者认证系列1
  20. 【5G NR】物理资源

热门文章

  1. 零代码实战 SharePoint 2013 BCS:三、实干篇
  2. joomla插件开发入门(六)
  3. 需求编写的几点经验之谈
  4. 鞋带的超炫系法-真是门艺术[转载]
  5. Spring配置文件详解三:Spring声明式事务管理
  6. GridView中实现单选RadioButton
  7. EasyUI-DataGrid之批量删除
  8. ASP.NET页面通过URL传递参数(一)(转载)
  9. 自定义ListView【通用】适配器并实现监听控件
  10. lnmp一键包的thinkphp5 nginx配置分享