树状数组一二三(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,上面就是这个本文的内容了,时间关系,我就不说那么多了,没办法,水一水活跃度。

相关内容

热门资讯

安卓系统包的预装软件,体验升级 你有没有发现,每次拿到新手机,打开安卓系统,总有一堆软件在那里等着你?这些软件就像小跟班一样,不管你...
安卓系统平板和win系统哪个好 你有没有想过,当你手捧一款平板,准备畅游知识的海洋时,是选择安卓系统还是Windows系统呢?这就像...
安卓系统如何下载立借,安卓系统... 你有没有想过,有时候资金周转不过来,急需一笔小钱?别急,今天就来教你怎么用安卓系统下载立借,轻松解决...
光遇ios系统如何加安卓系统,... 你有没有想过,你的光遇账号在iOS系统上玩得风生水起,但突然有一天,你想要在安卓系统上体验一番呢?别...
凤凰系统就是安卓吗,揭开其与安... 你有没有听说过凤凰系统?是不是觉得它和安卓有点像,但又不太一样?今天,我就来给你好好捋一捋,让你对凤...
电视系统安卓和云os,技术革新... 亲爱的读者们,你是否曾想过,家里的电视系统竟然也能如此智能?今天,就让我带你一起探索一下电视系统中的...
安卓系统基带工作原理,工作原理... 你有没有想过,你的安卓手机里那个默默无闻的基带,是怎么帮你打通世界的呢?它就像手机里的隐形英雄,每天...
锤子os安卓系统官网,创新与个... 你有没有听说过锤子OS安卓系统呢?这款系统可是近年来在手机圈里掀起了一股小热潮哦!今天,就让我带你一...
安卓值得玩的系统,盘点那些值得... 你知道吗?在手机世界里,安卓系统就像是个万能的魔法师,总能变出各种让人眼前一亮的玩法。今天,就让我带...
安卓系统如何app升级系统软件 你有没有发现,你的安卓手机最近是不是有点儿“慢吞吞”的?别急,这可能是你的手机在默默告诉你,是时候给...
电视盒子安卓系统设置,畅享智能... 亲爱的电视盒子用户们,你是否在享受高清影视的同时,也对安卓系统的设置感到一丝困惑呢?别担心,今天我就...
苹果13系统没有安卓好 你有没有发现,最近身边的朋友都在讨论苹果13的系统,说它没有安卓系统那么好。这可真是让人好奇,为什么...
vivo是安卓系统还是鸿蒙系统 你有没有想过,手机里的操作系统就像是我们的大脑,指挥着整个设备的运转呢?今天,咱们就来聊聊这个话题,...
王者安卓换苹果系统数据,数据迁... 你有没有想过,从王者安卓换到苹果系统,那数据迁移的过程,简直就像是一场穿越时空的冒险呢?想象你的英雄...
oppo安卓系统如何升级系统空... 亲爱的OPPO手机用户们,你是不是也遇到了这样的烦恼:想要升级安卓系统,却发现系统空间不足?别急,今...
安卓系统进不去无命令,安卓系统... 手机屏幕上突然黑屏了,安卓系统怎么就进不去了呢?别急,别慌,今天就来给你详细解析一下这个问题,让你轻...
适合安卓系统k歌软件,打造个人... 你有没有想过,在手机上也能尽情地唱出你的心声呢?现在,就让我带你走进一个神奇的世界,那就是适合安卓系...
安卓系统怎么充电的视频 手机电量告急,又到了充电的时刻啦!你是不是也和我一样,对安卓系统的充电方式充满了好奇?今天,就让我带...
ios系统与安卓系统的内存对比... 你有没有发现,手机里的世界越来越精彩了?各种应用层出不穷,游戏、社交、办公,样样都离不开手机。而支撑...
安卓7.1系统打开usb方法,... 你有没有想过,有时候你的安卓手机就像一个神秘的宝盒,里面藏着许多你意想不到的小秘密?今天,我就要给你...