博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
线段树
阅读量:4566 次
发布时间:2019-06-08

本文共 2180 字,大约阅读时间需要 7 分钟。

#define maxn 100007  //元素总个数  #define ls l,m,rt<<1  #define rs m+1,r,rt<<1|1  int Sum[maxn<<2],Add[maxn<<2];//Sum求和,Add为懒惰标记   int A[maxn],n;//存原数组数据下标[1,n]   //PushUp函数更新节点信息 ,这里是求和  void PushUp(int rt){Sum[rt]=Sum[rt<<1]+Sum[rt<<1|1];}  //Build函数建树   void Build(int l,int r,int rt){ //l,r表示当前节点区间,rt表示当前节点编号      if(l==r) {
//若到达叶节点 Sum[rt]=A[l];//储存数组值 return; } int m=(l+r)>>1; //左右递归 Build(l,m,rt<<1); Build(m+1,r,rt<<1|1); //更新信息 PushUp(rt); } void Update(int L,int C,int l,int r,int rt){
//l,r表示当前节点区间,rt表示当前节点编号 if(l==r){
//到叶节点,修改 Sum[rt]+=C; return; } int m=(l+r)>>1; //根据条件判断往左子树调用还是往右 if(L <= m) Update(L,C,l,m,rt<<1); else Update(L,C,m+1,r,rt<<1|1); PushUp(rt);//子节点更新了,所以本节点也需要更新信息 } void Update(int L,int R,int C,int l,int r,int rt){
//L,R表示操作区间,l,r表示当前节点区间,rt表示当前节点编号 if(L <= l && r <= R){
//如果本区间完全在操作区间[L,R]以内 Sum[rt]+=C*(r-l+1);//更新数字和,向上保持正确 Add[rt]+=C;//增加Add标记,表示本区间的Sum正确,子区间的Sum仍需要根据Add的值来调整 return ; } int m=(l+r)>>1; PushDown(rt,m-l+1,r-m);//下推标记 //这里判断左右子树跟[L,R]有无交集,有交集才递归 if(L <= m) Update(L,R,C,l,m,rt<<1); if(R > m) Update(L,R,C,m+1,r,rt<<1|1); PushUp(rt);//更新本节点信息 } void PushDown(int rt,int ln,int rn){ //ln,rn为左子树,右子树的数字数量。 if(Add[rt]){ //下推标记 Add[rt<<1]+=Add[rt]; Add[rt<<1|1]+=Add[rt]; //修改子节点的Sum使之与对应的Add相对应 Sum[rt<<1]+=Add[rt]*ln; Sum[rt<<1|1]+=Add[rt]*rn; //清除本节点标记 Add[rt]=0; } } int Query(int L,int R,int l,int r,int rt){
//L,R表示操作区间,l,r表示当前节点区间,rt表示当前节点编号 if(L <= l && r <= R){ //在区间内,直接返回 return Sum[rt]; } int m=(l+r)>>1; //下推标记,否则Sum可能不正确 PushDown(rt,m-l+1,r-m); //累计答案 int ANS=0; if(L <= m) ANS+=Query(L,R,l,m,rt<<1); if(R > m) ANS+=Query(L,R,m+1,r,rt<<1|1); return ANS; } //建树 Build(1,n,1); //点修改 Update(L,C,1,n,1); //区间修改 Update(L,R,C,1,n,1); //区间查询 int ANS=Query(L,R,1,n,1);

 

转载于:https://www.cnblogs.com/KasenBob/p/10282961.html

你可能感兴趣的文章
简单理解什么是递归(阶乘演示)
查看>>
http协议
查看>>
js倒计时,页面刷新时,不会从头计时
查看>>
洛谷1156垃圾陷阱
查看>>
python ==》 递归
查看>>
简单网络请求封装
查看>>
django —— MVT模型
查看>>
oracle 静默安装
查看>>
Python3基础(2)模块、数据类型及运算、进制、列表、元组、字符串操作、字典...
查看>>
服务器上centos 7 配置静态IP
查看>>
C# unsafe模式内存操作深入探索
查看>>
Redis拾遗(一)
查看>>
js字符串转换为Json对象的三种写法
查看>>
Is it possible to display icons in a PopupMenu?
查看>>
制作导航条
查看>>
iOS中的内存管理1
查看>>
23种设计模式全解析
查看>>
Learning Python 008 正则表达式-003 sub()方法
查看>>
要检测两个C文件的代码的抄袭情况
查看>>
iOS开发之应用内支付IAP全部流程
查看>>