博客
关于我
树状数组(单点修改区间查询、区间修改单点查询、区间修改区间查询)
阅读量:274 次
发布时间:2019-03-01

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

版权声明:本文为CSDN博主「Zars19」的原创文章,遵循CC 4.0 by-sa版权协议,转载请附上原文出处链接及本声明。

Description

 

分别对应的三道模板裸题: 

 

Solution

 

一个很优美的数据结构 

放图一张帮助日后忘记的时候回忆(图自百度百科) 

这里写图片描述

单点修改 区间查询

#include
#include
#include
#include
using namespace std;int n,m;int c[500005];int lowbit(int x){ return (-x)&x;}void add(int pos,int x){ while(pos<=n) { c[pos]+=x; pos+=lowbit(pos); }}void input(){ int x; for(int i=1;i<=n;i++) { scanf("%d",&x); add(i,x); }}int query(int pos){ int res=0; while(pos>0) { res+=c[pos]; pos-=lowbit(pos); } return res;}int main(){ scanf("%d%d",&n,&m); input(); int f,x,y; for(int i=1;i<=m;i++) { scanf("%d%d%d",&f,&x,&y); if(f==1) add(x,y); else if(f==2) cout<
<

区间修改 单点查询

#include
#include
#include
#include
using namespace std;int c[500005];int n,m;int lowbit(int x){ return x&(-x);}void add(int pos,int x){ while(pos<=n) { c[pos]+=x; pos+=lowbit(pos); }}int query(int pos){ int res=0; while(pos>0) { res+=c[pos]; pos-=lowbit(pos); } return res; } int main(){ scanf("%d%d",&n,&m); int x=0,y; for(int i=1;i<=n;i++) { scanf("%d",&y); add(i,y-x); x=y; } int opt,k; for(int i=1;i<=m;i++) { scanf("%d",&opt); if(opt==1) { scanf("%d%d%d",&x,&y,&k); add(x,k); add(y+1,-k); } else if(opt==2) { scanf("%d",&x); printf("%d\n",query(x)); } } return 0; }

区间修改 区间查询*

此处简略地说明一下 

原数组a,差分数组d 则有 
an=∑i=1ndian=∑i=1ndi 
所以 
∑i=1nai=∑i=1n∑j=1idj=∑i=1n(n−i+1)×di∑i=1nai=∑i=1n∑j=1idj=∑i=1n(n−i+1)×di 
=(n+1)×∑i=1ndi−∑i=1ndi×i=(n+1)×∑i=1ndi−∑i=1ndi×i 
于是我们维护两个树状数组,c1储存didi,c2储存di×i
 

#include
#include
#include
using namespace std;long long c[200005][2];int n,q;int lowbit(int x){ return x&(-x);}void add(int pos,int x,int f){ while(pos<=n) { c[pos][f]+=x; pos+=lowbit(pos); }}long long query(int pos,int f){ long long res=0; while(pos>0) { res+=c[pos][f]; pos-=lowbit(pos); } return res; }long long ask(int pos){ long long res=(pos+1)*query(pos,0)-query(pos,1); return res; }int main(){ scanf("%d",&n); int a=0,b,opt,x; for(int i=1;i<=n;i++) { scanf("%d",&b); add(i,b-a,0); add(i,(b-a)*i,1); a=b; } scanf("%d",&q); for(int i=1;i<=q;i++) { scanf("%d",&opt); if(opt==1) { scanf("%d%d%d",&a,&b,&x); add(a,x,0); add(b+1,-x,0); add(a,x*a,1); add(b+1,-x*(b+1),1); } else if(opt==2) { scanf("%d%d",&a,&b); printf("%lld\n",ask(b)-ask(a-1)); } } return 0;}

 

 

 

你可能感兴趣的文章
MSSQL日期格式转换函数(使用CONVERT)
查看>>
MSTP多生成树协议(第二课)
查看>>
MSTP是什么?有哪些专有名词?
查看>>
Mstsc 远程桌面链接 And 网络映射
查看>>
Myeclipse常用快捷键
查看>>
MyEclipse更改项目名web发布名字不改问题
查看>>
MyEclipse用(JDBC)连接SQL出现的问题~
查看>>
mt-datetime-picker type="date" 时间格式 bug
查看>>
myeclipse的新建severlet不见解决方法
查看>>
MyEclipse设置当前行背景颜色、选中单词前景色、背景色
查看>>
Mtab书签导航程序 LinkStore/getIcon SQL注入漏洞复现
查看>>
myeclipse配置springmvc教程
查看>>
MyEclipse配置SVN
查看>>
MTCNN 人脸检测
查看>>
MyEcplise中SpringBoot怎样定制启动banner?
查看>>
MyPython
查看>>
MTD技术介绍
查看>>
MySQL
查看>>
MySQL
查看>>
mysql
查看>>