树状数组一二三(Python版本)
创始人
2025-05-28 07:39:57
0

文章目录

  • 前言
  • lowbit操作
    • 数组划分
    • 操作
  • 单点修改,区间查询
  • 区间修改,点单查询
  • 区间修改,区间查询
  • 总结

前言

混个活跃度,本文内容主要为Python版本的树状数组,原理的话,这个不太好阐述(受限于博文篇幅的原因)不过这里还是会尽可能稍微解释一下,它里面的lowbit 操作的。

那么话说回来这个树状数组有什么用呢,其实主要就是为了对付这个区间问题,比如区间修改啥的。那么这些部分的操作有差分,线段树,树状数组啥的,那么今天要说的就是这个树状数组。
那么树状数组解决了哪些问题呢,大概可以解决这三种类型的问题:

  1. 区间查询, 单点修改
  2. 区间修改, 单点查询
  3. 区间修改, 区间查询

那么修改的是啥,查询的是啥呢。首先对于第一类问题,也就是使用树状数组最原始的问题,那么一开始,这个树状数组呢,其实是在平衡前缀和的功能,我们知道前缀和可以快速查询到当前位置i之前的和,这个时间复杂度是O(1) 的,但是当我们的原数组进行修改之后的话,我们的前缀数组就需要进行更新此时操作就是O(n)的,所以的话为了平衡这样的时间消耗,有这个树状数组,他的时间复杂度都是O(nlogn)的。

那么此时再配合差分,这样的话我们就可以实现这个区间修改的功能了。

lowbit操作

数组划分

那么整个树状数组的实现非常简单,那么其中一个比较重要的操作其实就是这个lowbit操作,那么这个操作的话其实和这个树状数组的原理有关。
首先我们知道一个数字(十进制)可以用二进制数字求和表示(我们的计算机就是这样操作的)因此对于一个长度为X的数组,我们可以通过二进制为进行划分。那么这里的话就要扯到这个玩意是如何划分的了:

假设一个数X=2^ki + 2^ki-1 + …+ 2^i
也就是一个数的二进制表示假设是这样的: 0100 1010
我们转换十进制的时候其实就是刚刚的公式
ki,ki-1 表示的都是在这个二进制表示当中为1的地方。

所以我们就可以这样把一个长度为X的区间,划分为(x-2^i, x],(x-2^i - 2^i2,x- 2^i]…
这样划分下去,那么显然lowbit的操作其实就是,找到最后的一个1的那个数b,然后X-b就可以划分出一个区间了。

那么此时我们的这个区间划分好了,我们的目的是为了实现这个区间查询和单点的修改。首先看到区间查询,我们要实现这个区间查询,然后区间被划分,所以的话我们就需要一个东西来记录原数组的值和区间的一个值。

红色的就是我们记录值的区间值的玩意。
在这里插入图片描述
我们用一个大数组来存储区间维护的信息。
那么此时
C[1] = nums[1]
C[2] = C[1]+A[2] = A[1] + A[2]

然后就有个规律:
C[i] = A[i-2^k +1] + …+ A[i]
那么这个k的话就是我们二进制表示里面为1的地方。那么我们的lowbit操作就是这样的,找到这个数(十进制)。

操作

那么现在我们就来看看这个lowbit的操作,这个的话就是这个这个公式:
从位运算的角度我们得到的公式是这样的:

def lowbit(x):return x&(~x+1)

然后的话,假设x是正数,~x 得到了x作为负数的反码,之后反码+1得到了-x的补码,然后知道我们的计算机是通过补码运算的,所以的话,我们的函数可以优化为

def lowbit(x);return x&(-x)

那么当x为负数的时候 x相当于正数的-x,-x相当于原来正数的x,所以情况是一样的。当然你也可以自己写一遍看看,记住计算机是补码运算就好了。

单点修改,区间查询

ok,我们现在就直接看到代码了,因为代码比较简单,就算你不理解也没关系,你只需要知道怎么用就好了。


class TreeList():def __init__(self,nums,maxn=10000):self.maxn = maxnself.c = [0]*self.maxnself.n = len(nums)for i in range(1,self.n+1):self.update(i,nums[i-1])def update(self,i,k):while(i<=self.n):self.c[i]+=ki+=self.lowbit(i)def getSum(self,i):# 1~i 之和ans = 0while(i>0):ans += self.c[i]i-=self.lowbit(i)return ansdef getSumLR(self,L,R):ans = 0ans+=self.getSum(R)ans-=self.getSum(L-1)return ansdef lowbit(self,x):return x&(-x)

区间修改,点单查询

加了个差分数组

class TreeList2():def __init__(self,nums,maxn=10000):self.maxn = maxnself.nums = [0]+numsself.c = [0]*self.maxnself.n = len(nums)self.d = [0]*self.maxnfor i in range(1,self.n+1):self.d[i] = self.nums[i]-self.nums[i-1]self.insert(i,self.d[i])def insert(self,i,k):while(i<=self.n):self.c[i]+=ki+=self.lowbit(i)def update(self,l,r,k):self.insert(l,k)self.insert(r+1,-k)def get(self,i):ans = 0while(i>0):ans += self.c[i]i-=self.lowbit(i)return ansdef lowbit(self,x):return x&(-x)

区间修改,区间查询

其实这个区间查询当L=R的时候,就是一个单点查询。

class TreeList3():def __init__(self,nums,maxn=10000):self.maxn = maxnself.nums = [0]+numsself.c = [0]*self.maxnself.n = len(nums)self.b = [0]*self.maxnfor i in range(1,self.n+1):self.insert(i,self.nums[i]-self.nums[i-1])def insert(self,x,k):i = xwhile(i<=self.n):self.b[i]+=kself.c[i]+=x*ki+=self.lowbit(i)def update(self,l,r,k):self.insert(l,k)self.insert(r+1,-k)def getSum(self,x):ans = 0i = xwhile(i>0):ans += (x+1)*self.b[i] - self.c[i]i-=self.lowbit(i)return ansdef getSumLR(self,L,R):ans = self.getSum(R)ans -= self.getSum(L-1)return ansdef lowbit(self,x):return x&(-x)

总结

ok,上面就是这个本文的内容了,时间关系,我就不说那么多了,没办法,水一水活跃度。

相关内容

热门资讯

安卓如何操控苹果系统,揭秘跨平... 你知道吗?在这个科技飞速发展的时代,安卓和苹果两大操作系统之间的较量可是从未停歇。虽然它们各自有着忠...
安卓系统账户同步数据,畅享无缝... 你有没有遇到过这种情况:手机里存了那么多宝贝照片、重要文件,结果换了个新手机,却发现那些宝贝全都不翼...
安卓系统不停推送广告,安卓系统... 你有没有发现,最近你的安卓手机是不是越来越“热情”了?没错,就是那个不停在你屏幕上跳来跳去的广告!今...
airpods可以和安卓系统,... 你有没有想过,那些炫酷的AirPods竟然也能和安卓手机完美搭配?没错,就是那个我们平时只听说和iP...
安卓系统实体键盘不对,创新与挑... 你是不是也遇到了这个问题?安卓手机的实体键盘突然不对劲了,按下去没反应,或者反应迟钝,简直让人抓狂!...
汽车导航改装安卓系统,安卓系统... 你有没有想过,你的汽车导航系统是不是已经out了?现在,让我来给你揭秘如何给你的爱车来一次科技大变身...
安卓系统如何限制下载,安卓系统... 你有没有发现,手机里的安卓系统越来越智能了?不过,这也意味着有时候我们不小心就会下载一些不想要的软件...
安卓系统调成日语,概要の副標題... 你有没有想过,你的安卓手机竟然可以变成一个日式小天地呢?没错,就是那种动漫里常见的日语界面,是不是听...
男生耳机推荐安卓系统,男生耳机... 耳机可是现代生活中不可或缺的小玩意儿,尤其是对于喜欢听音乐的男生来说,一副好耳机简直就是灵魂的伴侣。...
安卓同版本升级系统,功能优化与... 你知道吗?最近手机界可是热闹非凡呢!各大品牌纷纷推出了安卓同版本升级系统,让我们的手机焕然一新。今天...
安卓更换别的手机系统,轻松切换... 你有没有想过,你的安卓手机用久了,是不是有点审美疲劳了呢?或者,你最近是不是对其他手机系统产生了浓厚...
安卓系统单机神雕侠侣,指尖重温 你有没有想过,在手机上也能体验一把江湖恩怨、侠骨柔肠?没错,就是那个让人心驰神往的《神雕侠侣》!今天...
安卓系统键盘语言切换,安卓系统... 你有没有发现,手机上的安卓系统键盘语言切换功能,简直就像是个神奇的魔法棒,轻轻一点,就能让文字飞舞在...
oppok1安卓系统,性能与体... 你有没有发现,最近手机圈里又掀起了一股热潮?没错,就是OPPO K1这款新机!这款手机不仅外观时尚,...
安卓系统环境的搭建,从零开始构... 想要在电脑上体验安卓系统的魅力,是不是已经跃跃欲试了呢?别急,今天就来手把手教你如何搭建一个属于自己...
【MySQL】锁 锁 文章目录锁全局锁表级锁表锁元数据锁(MDL)意向锁AUTO-INC锁...
【内网安全】 隧道搭建穿透上线... 文章目录内网穿透-Ngrok-入门-上线1、服务端配置:2、客户端连接服务端ÿ...
GCN的几种模型复现笔记 引言 本篇笔记紧接上文,主要是上一篇看写了快2w字,再去接入代码感觉有点...
数据分页展示逻辑 import java.util.Arrays;import java.util.List;impo...
Redis为什么选择单线程?R... 目录专栏导读一、Redis版本迭代二、Redis4.0之前为什么一直采用单线程?三、R...