树状数组维护区间和的模型及其拓广的简单总结

 

by wyl8899   

 

树状数组的基本知识已经被讲到烂了,我就不多说了,下面直接给出基本操作的代码。

假定原数组为a[1..n],树状数组b[1..n],考虑灵活性的需要,代码使用int *a传数组。

#define lowbit(x) ((x)&(-(x)))int sum(int *a,int x){int s=0;for(;x;x-=lowbit(x))s+=a[x];return s;}void update(int *a,int x,int w){for(;x<=n;x+=lowbit(x))a[x]+=w;}

sum(x)返回原数组[1,x]的区间和,update(x,w)将原数组下标为x的数加上w。

这两个函数使用O(操作数*logn)的时间和O(n)的空间完成单点加减,区间求和的功能。

 

接下来做一些升级,让树状数组完成区间加减,单点查询的功能。

直接做的话很困难,需要对问题做一些转化。

考虑将原数组差分,即令d[i]=a[i]-a[i-1],特别地,d[1]=a[1]。

此时a[i]=d[1]+..+d[i],所以单点查询a[i]实际上就是在求d数组的[1..i]区间和。

而区间[l,r]整体加上k的操作,可以简单地使用d[l]+=k和d[r+1]-=k来完成。

于是,我们用树状数组来维护d[],就可以解决问题了。

 

下面再升级一次,完成区间加减,区间求和的功能。

仍然沿用d数组,考虑a数组[1,x]区间和的计算。d[1]被累加了x次,d[2]被累加了x-1次,...,d[x]被累加了1次。

因此得到

sigma(a[i])

=sigma{d[i]*(x-i+1)}

=sigma{ d[i]*(x+1) - d[i]*i }

=(x+1)*sigma(d[i])-sigma(d[i]*i)

所以我们再用树状数组维护一个数组d2[i]=d[i]*i,即可完成任务。

POJ 3468就是这个功能的裸题,下面给出代码。

[请注意我们上面的讨论都假定了a[]初始全是0。如果不是这样呢?下面的程序里给出了一个相对简便的处理办法。]

// POJ 3468   Using BIT#include <cstdio>const int maxn=100010;__int64 a[maxn],b[maxn],c[maxn];int n,m;inline int lowbit(const int &x){return x&(-x);}__int64 query(__int64 *a,int x){__int64 sum=0;while(x){sum+=a[x];x-=lowbit(x);}return sum;}void update(__int64 *a,int x,__int64 w){while(x<=n){a[x]+=w;x+=lowbit(x);}}int main(){int l,r,i;__int64 ans,w;char ch;scanf("%d%d",&n,&m);a[0]=0;for(i=1;i<=n;++i){scanf("%I64d",&a[i]);a[i]+=a[i-1];}while(m--){scanf("%c",&ch);while(ch!='Q' && ch!='C')scanf("%c",&ch);if(ch=='Q'){scanf("%d%d",&l,&r);ans=a[r]-a[l-1]+(r+1)*query(b,r)-l*query(b,l-1)-query(c,r)+query(c,l-1);printf("%I64d\n",ans);}else{scanf("%d%d%I64d",&l,&r,&w);update(b,l,w);update(b,r+1,-w);update(c,l,w*l);update(c,r+1,-(r+1)*w);}}return 0;}

[当a[]初始不全0的时候,我们就只维护后来加上去的部分,查询区间和的时候再补上初始的时候这一段的区间和就可以了。]

======================一维到二维的分割线=========================

接下来到二维树状数组。

先看看sum和update变成什么样子了吧。

inline int gs(int a[maxn][maxn],int x,int y){int s=0,t;for(;x;x-=lowbit(x))for(t=y;t;t-=lowbit(t))s+=a[x][t];return s;}inline void gp(int a[maxn][maxn],int x,int y,int w){int t;for(;x<=n;x+=lowbit(x))for(t=y;t<=m;t+=lowbit(t))a[x][t]+=w;}

gs就是sum,gp就是update,由于需要多次调用的缘故,改成了更短的名字。

单点加减,矩形求和并不难,直接用上面的两段就行了。

需要注意的是矩形的求和怎么求。上面的代码返回的是(1,1)-(x,y)矩形的和。

那么(x1,y1)-(x2,y2)的矩形和由下式给出:

sum(x2,y2)-sum(x1-1,y2)-sum(x2,y1-1)+sum(x1-1,y1-1)

画个图就很好理解了。

 

对于涉及矩形加减的情形,我们发现一维中的差分的办法在二维的情况用不出来,所以要改一下。思考一下一维中的差分的另外一个含义:d[i]同时也表示d[i..n]的整体增量,d[i]+=k就意味着把d[i]..d[n]全部加上了k。理解了之后就发现这个意义上可以推广到二维,仍假设原矩形初始全为0,以便接下来的叙述。

令a[x,y]表示(x,y)-(n,m)矩形的整体增量,其中(n,m)是边界。

那么(x1,y1)-(x2,y2)矩形整体加k的代码就是

gp(a,x1,y1,w); gp(a,x2+1,y1,-w);gp(a,x1,y2+1,-w); gp(a,x2+1,y2+1,w);

仍然是建议画个图来帮助理解。

 

至此,矩形加减,单点查询的问题得到了解决。

 

重头戏在这里,矩形加减,矩形求和。

求原矩形(1,1)-(x,y)的和,结果由下式给出

sigma(i=1..x,j=1..y) a[i,j]*(x-i+1)*(y-j+1)

很好理解吧? 但是这个式子并不是那么容易求和的,展开一下求和的部分得到

a[i,j]*  ( (x+1)(y+1) - (x+1)*j - (y+1)*x + i*j )

整个式子就是

(x+1)(y+1)sigma(a[i,j]) - (x+1)sigma(a[i,j]*j) - (y+1)sigma(a[i,j]*i) + sigma(a[i,j]*i*j)

知道怎么处理了吧?如果没有请回去复习一维的处理方法。

令b[i,j]=a[i,j]*i  c[i,j]=a[i,j]*j  d[i,j]=a[i,j]*i*j

维护a,b,c,d一共四个二维树状数组,问题得到解决。

tyvj p1716就是实现这两个功能的裸题,下面给出完整代码。


 

// tyvj p1716  using 2D BIT#include<cstdio>#include<cstring>#define lowbit(x) ((x)&(-(x)))const int maxn=2049;int a[maxn][maxn],b[maxn][maxn],c[maxn][maxn],d[maxn][maxn];int n,m;inline int gs(int a[maxn][maxn],int x,int y){int s=0,t;for(;x;x-=lowbit(x))for(t=y;t;t-=lowbit(t))s+=a[x][t];return s;}inline void gp(int a[maxn][maxn],int x,int y,int w){int t;for(;x<=n;x+=lowbit(x))for(t=y;t<=m;t+=lowbit(t))a[x][t]+=w;}inline int sum(int x,int y){return (x+1)*(y+1)*gs(a,x,y)-(y+1)*gs(b,x,y)-(x+1)*gs(c,x,y)+gs(d,x,y);}inline void update(int x1,int y1,int x2,int y2,int w){gp(a,x1,y1,w); gp(a,x2+1,y1,-w);gp(a,x1,y2+1,-w); gp(a,x2+1,y2+1,w);gp(b,x1,y1,w*x1); gp(b,x2+1,y1,-w*(x2+1));gp(b,x1,y2+1,-w*x1); gp(b,x2+1,y2+1,w*(x2+1));gp(c,x1,y1,w*y1); gp(c,x2+1,y1,-w*y1);gp(c,x1,y2+1,-w*(y2+1)); gp(c,x2+1,y2+1,w*(y2+1));gp(d,x1,y1,w*x1*y1); gp(d,x2+1,y1,-w*(x2+1)*y1);gp(d,x1,y2+1,-w*x1*(y2+1)); gp(d,x2+1,y2+1,w*(x2+1)*(y2+1));}int main(){int x1,y1,x2,y2,w;char ch;scanf("%c",&ch);while(ch!='X')scanf("%c",&ch);scanf("%d%d\n",&n,&m);memset(a,0,sizeof(a));memset(b,0,sizeof(b));memset(c,0,sizeof(c));memset(d,0,sizeof(d));while(scanf("%c",&ch)!=EOF){scanf("%d%d%d%d",&x1,&y1,&x2,&y2);if(ch=='L'){scanf("%d\n",&w);update(x1,y1,x2,y2,w);}else{scanf("\n");printf("%d\n",sum(x2,y2)-sum(x1-1,y2)-sum(x2,y1-1)+sum(x1-1,y1-1));}}return 0;}

wy18899

没原文链接,只能写个原创,但是不是我写的

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/445335.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Python(6)-算数运算符

算数运算符1.算数运算符2.优先级1.算数运算符 加 减- 乘* 除/ 取商// 取余数% 幂**(能算n次方&#xff1a; 2**38&#xff0c;一直以为只能算平方) 扩展&#xff1a; 乘法用于字符串&#xff1a;字符串重复指定的次数&#xff0c;要拼接的次数很长时&#xff0c;用乘号很方便…

推荐算法--其他信息(07)

文章目录目录1.利用上下文信息1.1时间上下文1.2地点上下文2.利用网络社交数据2.1 获取网络社交数据途径2.2 社交网络数据2.3 基于社交网络的推荐2.4 推荐算法2.5 给用户推荐好友目录 1.利用上下文信息 1.1时间上下文 用户的兴趣是随着时间变化的&#xff0c;三天打鱼两天晒网…

动态规划的深入探讨

一、引言 动态规划是一种重要的程序设计思想&#xff0c;具有广泛的应用价值。使用动态规划思想来设计算法&#xff0c;对于不少问题往往具有高时效&#xff0c;因而&#xff0c;对于能够使用动态规划思想来解决的问题&#xff0c;使用动态规划是比较明智的选择。 能够用动态规…

Python(7)-程序执行的原理

程序执行的原理1.计算机中的三个核心部件2.程序执行的原理3.程序的作用1.计算机中的三个核心部件 CPU&#xff1a;中央处理区&#xff0c;超大规模的集成电路&#xff0c;负责处理数据、计算 内存&#xff1a;临时存储数据&#xff0c;断电数据消失&#xff0c;读取数据快 硬盘…

推荐系统读书笔记(推荐系统实战)

随着信息技术和互联网的发展&#xff0c;人们逐渐从信息匮乏的时代走入了信息过载的时代。在这个时代&#xff0c;无论是信息消费者还是信息生产者都遇到很大的挑战&#xff1b;对于消费者&#xff0c;从大量信息中找到自己感兴趣的信息是一件非常困难的事情&#xff1b;对于信…

橙白oj 2017级《算法分析与设计》-练习02

注&#xff1a;A题我以为给新生出的&#xff0c;应该贼简单&#xff0c;是按顺序消灭&#xff0c;卡了十几分钟&#xff0c;成了最后一个ac的题&#xff0c;真是菜的真实。 Problem A: Description 白细胞是人体与疾病斗争的“卫士”。当病菌侵入人体体内时&#xff0c;白细胞…

python(9)-变量、input函数

变量、input函数1.变量的定义(不可变对象)2.变量的类型3.变量的命名规范4. 不同类型的数据计算5.类型转换函数6.input()7.Tips程序是用来处理数据的&#xff0c;而变量是用来存储数据的。 关于函数&#xff0c;是一个提前准备好的代码&#xff1b;可以直接使用&#xff0c;不用…

推荐算法--总结(08)

一、推荐系统结构二、推荐引擎算法&#xff08;Algorithm&#xff09;1、协同过滤推荐算法1.1 关系矩阵与矩阵计算1.1.1 用户与用户&#xff08;U-U矩阵&#xff09;1.1.2 物品与物品&#xff08;V-V矩阵&#xff09;1.1.3 用户与物品&#xff08;U-V矩阵&#xff09;1.1.4 奇异…

算法总结-1算法入门

1.0 前言 算法&#xff08;Algorithm&#xff09;是指解题方案的准确而完整的描述&#xff0c;是一系列解决问题的清晰指令&#xff0c;算法代表着用系统的方法描述解决问题的策略机制。也就是说&#xff0c;能够对一定规范的输入&#xff0c;在有限时间内获得所要求的输出。 …

Python(10)- 格式化输出%

格式化输出1. 格式化输出浮点数2. 格式化输出整数3. 格式化输出浮点数4. 格式化输出百分号%变量的格式化输出&#xff1a;input函数将信息输出到控制台&#xff0c;实现变量和文字一起输出。 %格式化操作符&#xff0c;和不同的字符连用&#xff0c;%s 字符串&#xff0c;%d 十…

FM系列算法解读(FM+FFM+DeepFM)

在计算广告中&#xff0c;CTR是非常重要的一环。对于特征组合来说&#xff0c;业界通用的做法主要有两大类&#xff1a;FM系列和Tree系列。这里我们来介绍一下FM系列。   在传统的线性模型中&#xff0c;每个特征都是独立的&#xff0c;如果需要考虑特征与特征之间的相互作用…

二叉树层序遍历

层序遍历序列为&#xff1a;ABCDEFG 思路&#xff1a;栈是先进后出的数据结构&#xff0c;而队列是先进先出的数据结构。 我们层序遍历&#xff0c;很明显&#xff0c;先遇到的节点先打印&#xff0c;不同于前中后序遍历&#xff0c;我们采用队列结构。 具体执行过程如下&…

深度学习(01)-- 基础学习

文章目录目录1. 深度学习基础1.1 深度学习总览1.2 深度网络训练过程1.2.1 传统神经网络的训练方法为什么不能用在深度神经网络1.2.2 deep learning训练过程1.3 数学知识&#xff1a;2. 九种深度学习模型2.1 受限玻尔兹曼机RBM2.2 自编码器AE&#xff08;降维&#xff09;2.3 深…

MachineLearning(1)-激活函数sigmoid、损失函数MSE、CrossEntropyLoss

损失函数1.激活函数2.损失函数2.1均方误差损失函数2.2交叉熵损失函数2.3 NLLLoss()2.4 BCELoss()1.激活函数 全连接网络又叫多层感知器&#xff0c;多层感知器的基本单元神经元是模仿人类神经元兴奋与抑制机制&#xff0c;对其输入进行加权求和&#xff0c;若超过某一阈值则该…

Java的IO总结

非流式文件类--File类 从定义看&#xff0c;File类是Object的直接子类&#xff0c;同时它继承了Comparable接口可以进行数组的排序。 File类的操作包括文件的创建、删除、重命名、得到路径、创建时间等&#xff0c;以下是文件操作常用的函数。 File类是对文件系统中文件以及文…

深度学习(02)-- ANN学习

文章目录目录1.神经网络知识概览1.1深度学习顶会1.2相关比赛1.3神经网络知识概览1.4神经网络编程一般实现过程2.简单神经网络ANN2.1 数据集&#xff1a;2.2 网络结构&#xff1a;2.3 代码实现2.3.1 读取数据&#xff0c;并做处理2.3.2 构建网络结构2.3.3 训练网络目录 1.神经网…

python(11)-if语句,断言assert

分支语句if1.if基本语法2 if语句的嵌套3 比较运算符号4 逻辑运算符&#xff1a;5 整数随机数初应用6 tip7.断言assert1.if基本语法 if语句开发中的应用场景&#xff1a;如果条件成立做一件事情&#xff0c;如果条件不成立做另外一件事情。有了if语句&#xff0c;程序有了分支.…

深度学习(03)-- CNN学习

文章目录目录1.CNN学习2.Keras深度学习框架目录 1.CNN学习 卷积神经网络CNN总结 从神经网络到卷积神经网络&#xff08;CNN&#xff09;我们知道神经网络的结构是这样的&#xff1a; 那卷积神经网络跟它是什么关系呢&#xff1f;其实卷积神经网络依旧是层级网络&#xff0c;…

Collections 工具类常见方法

Collections 工具类常用方法: 排序查找,替换操作同步控制(不推荐&#xff0c;需要线程安全的集合类型时请考虑使用 JUC 包下的并发集合) 排序操作 void reverse(List list)//反转 void shuffle(List list)//随机排序 void sort(List list)//按自然排序的升序排序 void sort(L…