博客
关于我
树状数组(单点修改区间查询、区间修改单点查询、区间修改区间查询)
阅读量: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;}

 

 

 

你可能感兴趣的文章
Nacos在双击startup.cmd启动时提示:Unable to start embedded Tomcat
查看>>
Nacos如何实现Raft算法与Raft协议原理详解
查看>>
Nacos安装教程(非常详细)从零基础入门到精通,看完这一篇就够了
查看>>
nacos注册失败,Feign调用失败,feign无法注入成我们的bean对象
查看>>
nacos源码 nacos注册中心1.4.x 源码 nacos源码如何下载 nacos 客户端源码下载地址 nacos discovery下载地址(一)
查看>>
Nacos编译报错NacosException: endpoint is blank
查看>>
NACOS部署,微服务框架之NACOS-单机、集群方式部署
查看>>
Nacos配置中心集群原理及源码分析
查看>>
nacos配置自动刷新源码解析
查看>>
Nacos集群搭建
查看>>
nacos集群搭建
查看>>
nagios安装文档
查看>>
name_save matlab
查看>>
Nami 项目使用教程
查看>>
NAT-DDNS内网穿透技术,解决动态域名解析难题
查看>>
NativePHP:使用PHP构建跨平台桌面应用的新框架
查看>>
NAT技术
查看>>
NAT模式下虚拟机centOs和主机ping不通解决方法
查看>>
NAT的两种模式SNAT和DNAT,到底有啥区别?
查看>>
Navicat for MySQL 命令列 执行SQL语句 历史日志
查看>>