HDU 1166 敌兵布阵 树状数组小结(更新)

时间:2021-09-04 14:15:27

树状数组(Binary Indexed Tree(BIT), Fenwick Tree)

是一个查询和修改复杂度都为log(n)的数据结构。主要用于查询任意两位之间的所有

元素之和,但是每次只能修改一个元素的值;经过简单修改可以在log(n)的复杂度下进行范围修改,但是这时只能查询其中一个元素的值。

这种数据结构(算法)并没有C++和Java的库支持,需要自己手动实现。在Competitive Programming的竞赛中被广泛的使用。树状数组

和线段树很像,但能用树状数组解决的问题,基本上都能用线段树解决,而线段树能解决的树状数组不一定能解决。相比较而言,树状数组

效率要高很多。但使用范围比线段树小(如查询每个区间最小值问题需要线段树)。

分析上面的几组式子可知,当 i 为奇数时,ci=ai ;当 i 为偶数时,就要看 i 的因子中最多有二的多少次幂,

例如,6 的因子中有 2 的一次幂,等于 2 ,所以 c6=a5+a6(由六向前数两个数的和),4 的因子中有 2 的两次幂,等于 4

所以 c4=a1+a2+a3+a4(由四向前数四个数的和)。

HDU 1166 敌兵布阵  树状数组小结(更新)

(重点操作)定义问题如下:我们有n个盒子,可能的操作为:

  1. 修改操作:往第i个盒子增加石子
  2. 计算一段区间和:计算第k个盒子到第l个盒子的石子数量(包含第k个和第l个)

原始的解决方案中(即用普通的数组进行存储,box[i]存储第i个盒子装的石子数), 操作1和操作2的时间复杂度分别是O(1)和O(n)。假如我们进行m次操作,最坏情况下, 即全为第2种操作,时间复杂度为O(n*m)。使用数状数组,它在最坏情况下的时间复杂度也为O(m log n),但比起RMQ, 它更容易编程实现,并且所需内存空间更少。

下面来对树状数组的基本思想和操作进行阐述:

idx记为BIT(树状数组)的索引,r记为idx二进制表示中最右边的1后面0的个数(这个r后面还会出现记得它的含义), 比如idx=1100(即十进制的12),那么r=2。tree[idx]记为f数组中, 索引从(idx-2r +1)到idx的所有数的和,包含f[idx-2r +1]和f[idx]。即: tree[idx]=f[idx-2r +1]+…+f[idx],对照表1.1和1.2理解和推算一下,你就会一目了然。 我们也可称idx对索引(idx-2r +1)到索引idx负责(比如表1.2的 8管着1-8,9管着9)。

这里先补充一句。每个整数都能表示为一些2的幂次方的和,比如13,其二进制表示为1101,所以它能表示为: 13 = 20 + 22 + 23

HDU 1166 敌兵布阵  树状数组小结(更新)

HDU 1166 敌兵布阵  树状数组小结(更新)

假设我们要得到索引为13的累积频率(即c[13]),在二进制表示中,13=1101。因此(你看一下上面那个图), 我们是不是可以这样计算:c[1101]=tree[1101]+tree[1100]+tree[1000]  (c[13]=tree[13](管着f[13])+tree[12](管着f[9]~f[12])+tree[8](管着f[1]~f[8]))

先观察上面这个式子 c[1101]=tree[1101]+tree[1100]+tree[1000]    有没有发现tree[ ]有上面规律?

分离出最后的1


这里先给出上面那个问题的规律;每次分离出最后的1

注意: 最后的1表示一个整数的二进制表示中,从左向右数最后的那个1。

我们知道,对一个数取负等价于对该数的二进制表示取反加1

num & -num = a1b & a^- 1b = (0…0)1(0…0)

(直接记住!!!)

 int lowbit(int x)

 {

     return x&(-x);

 }

记得刚刚那个r(ridx的二进制表示中最右边的1后面0的个数))两者其实是一样的  即 lowbit(i) ==2^r

读取累积频率


给定索引idx,如果我们想获取累积频率即c[idx](即从1idx中所有元素的总和),我们只需初始化sum=0, 然后当idx>0时,重复以下操作:sum加上tree[idx], 然后将idx减去相应的lowbit(idx)。

 int read(int idx){

        int sum=;

        while(idx>){

               sum+=tree[idx];

               idx -= lowbit(idx);

        }

        return sum;

 }

为什么可以这么做呢?(重点)

关键是tree数组设计得好。我们知道,tree数组是这么定义的: tree[idx] = f[idx-2r +1] +…+ f[idx].

上面的程序sum加上tree[idx]后,去掉idx最后的1(比如1100变成1000),假设变为idx1,

那么有idx1 = idx-2r (这里的r表示0的个数,上面说过,比如12(1100)去掉12最后的1(即减掉100(4))变成8(1000)), 

sum接下来加上tree[idx1] = f[idx1-2r1 +1] +…+ f[idx1] = f[idx1-2r1 +1] +…+f[idx-2r ], 我们可以看到tree[idx1]表达示的最右元素为f[idx-2r ]

这与tree[idx]表达式的最左元 素f[idx-2r +1]无缝地连接了起来。所以,只需要这样操作下去,即可求得f[1]+…+ f[idx],即c[idx]的结果。

来看一个具体的例子,当idx=13时,初始sum=0:

tree[1101]=f[13]

tree[1100]=f[9]+...+f[12]   (利用上面的解法发现f[12]与f[13]连接在一起)

tree[1000]=f[1]+...+f[8] (利用上面的解法发现f[8]与f[9]连接在一起)

c[1101]=f[1]+...+f[13]=tree[1101]+tree[1100]+tree[1000]

还记得刚才我们讲的。每个整数都能表示为一些2的幂次方的和,比如13,其二进制表示为1101,所以它能表示为:

13 = 20 –>13+ 22–>12,11,10,9 + 23–>8,7,6,5,4,3,2,1

改变某个位置的频率并且更新数组


当我们改变f数组中的某个值,比如f[idx],那么tree数组中哪些元素需要改变呢? 在取累积频率一节,我们每累加一次tree[idx],就将idx最后一个1移除, 然后重复该操作。而如果我们改变了f数组,比如f[idx]增加val,我们则需要为当前索引的 tree数组增加val:tree[idx] += val。然后idx更新为idx加上其最后的一个1, 当idx不大于MaxVal时,不断重复上面的两个操作。

HDU 1166 敌兵布阵  树状数组小结(更新)

这里做一个简单地理解,还以13为例,1101,首先它只管1个数,由于没有右边的0,但是它的上面的idx肯定有1后面10的那么它就可以分管其自身和它前面一个数,这就是14,就像下面的示例图

1101   13

+    1

1110   14

那么加上一个lowbit(13)其实就是加了一个一倍的最小可以管理的个数

接下来,就应该加上2了,由于1后面一个0,变成16,然后继续加。

其实可以看得出加的是一个相同的个数,让其翻倍,以便让上面管理到它的数组修改值。

 void update(int idx,int val){

        while(idx<=n){

               tree[idx]+=val;

               idx+=lowbit(idx);

        }

 }

下面是hdu1166的树状数组的代码:

需要注意输入输出要用C的输入输出,如果用C++的输入输出流会超时

 /*树状数组模板题*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define MAXN 50005
using namespace std;
int que[MAXN],c[MAXN];
char chr[];
int lowbit(int x)
{
return x&(-x);//用来计算a^k中的k
}
//实现每个树状数组元素所分管的元素之和的计算
void create(int n)
{
int i,s,j;
for(i=;i<=n;i++)
{
s=lowbit(i);
for(j=;j<s;j++)//含自身向前推s个
{
c[i]=c[i]+que[i-j];
}
}
}
//求数组的和的算法如下
int sum(int n)
{
int s=;
while(n>)
{
s+=c[n];
n=n-lowbit(n);
}
return s;
}
//修改操作,往上修改
void change(int i,int n,int x)
{
while(i<=n)
{
c[i]=c[i]+x;
i=i+lowbit(i);
}
}
int main()
{
int t,cas=,i,j,n,x,s,a,b;
cin>>t;
while(t--)
{
memset(c,,sizeof(c));
cin>>n;
for(i=;i<=n;i++)
scanf("%d",&que[i]);
create(n);
cout<<"Case "<<cas++<<":"<<endl;
while()
{
scanf("%s",chr);
if(!strcmp(chr,"Add"))
{
scanf("%d%d",&j,&x);
change(j,n,x);
}
else if(!strcmp(chr,"Sub"))
{
scanf("%d%d",&j,&x);
change(j,n,-x);
}
else if(!strcmp(chr,"Query"))
{
scanf("%d%d",&a,&b);
s=sum(b)-sum(a-);
printf("%d\n",s);
}
else
break;
}
}
return ;
}