【算法】树状数组

算法适用于求前k个整数的和，一般的做法是设置一个sum数组来表示

1、问题升级一下：在查询的过程中随时给第x个整数加上一个整数v，同样要求查询第K个整数的和。此时若还是原来的sum数组，则要遍历来更新，复杂度为O(N)，而用树状数组则时间要更快。

2、再来一题：计算序列中在元素左边比该元素小的元素个数。可以用hash(即map)来做，也可用树状数组：

update(x,1);
cout<<getSum(x-1);

方便吧~~~~

下面上树状数组的代码：

int lowbit(int x)//返回能整除x的最大2的幂次（例如6的lowbit(6)为2）
{
	return x&(-x);
}
int a[100];//原数组
int c[100];//树状数组，表示从i往前lowbit(i)个整数的和
		  //下标必须从1开始，c[i]的覆盖长度为lowbit(i)
int getSum(int x)//返回前x个整数之和
{
    int sum = 0;
    for(int i = x; i > 0;i -= lowbit(i))//注意是i > 0不是>=
        sum += c[i];
    return sum;
}
void update(int x, int v)//将第x个整数加上v
{
    for(int i = x;i < 100001; i += lowbit(i))//写成i<=n也可
        c[i] += v;
}

int main()
{	
	//初始化数组c
	for(int i = 1 ; i <= n ; i++)
    {
        int x;cin>>x;
        update(i,x);
    }
}

若要求[x,y]之间内的整数和，只需getSum(y) - getSum(x-1)即可

若不好理解，这里上个图：

那么当数很大的时候如10^9次方，A[i]<=N不成立，树状数组开不了那么大就没法使用了呢？不是的哦，例如{9999999999,18,66,88888}，如果只需考虑他们之间大小的关系，我们可以把序列看成{2，5，1，3，4}，其实是等价的。这种技巧成为离散化。

下面上个可以用树状数组解的PAT甲级题：

题目：1057 Stack (30分)