【专题】树状数组_树状数组专题-CSDN博客

本文链接：https://blog.csdn.net/mmy1996/article/details/53091360

树状数组

树状数组可以快速地求取连续地字串和

给定一个n个元素的数组 $A_1、A_2...A_n$ ,树状数组支持以下两种操作。
$Add(x,d):让A_x增加d$
$Query(L,R):计算A_L+A_{L+1}...A_R$

【例题一】

LA4329pingpong

题目大意

给出n个数 $a_1、a_2...a_n（1\le a_i \le 100000）$ ,设 $a_i$ 之前比 $a_i$ 小的数的个数为 $C_i$ ，求 $\sum_{i=1}^n C_i$

分析

现在先考虑一个与此题无关的问题，给定一个数组，设计一种方法支持一种查询操作：查询这个数组中比x小的数的个数。这里x是可变的。一种做法是先排一次序，再遍历一次就可以了。复杂度是 $O(nlgn)$ ，观察发现这里 $a_i$ 是个不大的数，为了节省时间，排序也可以不用了（或者叫计数排序）。建一个数组，统计出值为m的数的个数存入数组中下标为m的地方。这样也可以达到类似排序的效果。最后再遍历一次，累加数组中的值（到下标为x时停止）
再回到上面的问题，这里的区间是变化的，但仅仅是从前面的区间上增加了元素，所以还是可以用上个区间的计数结果进行更新就行。但对于每一个 $a_i$ ，在求 $a_i$ 之前的区间里比 $a_i$ 小的数的个数时都用去遍历累加一遍会太慢，而这里，树状数组就刚好是为了这种查询而设计的。
这个问题和求一组数的逆序对数几乎一样，只是把小改成大。

【例题二】

poj 2352 Stars

题目大意

n个星星，坐标 $(x,y)$ ，设第i个星左下方(包括边界)星星的数量为z，则称level为x。求出所有level水平的数量。输入按照先y递增，y相同则x递增。

分析

初一看像是一个二维问题，但其实一维就可以解决。输入是先y递增，y相同则x递增。这个条件其实很重要。问题可以等价于求每个星左下方星星的数量。按照输入的顺序进行处理。可以发现对于当前要处理的星，之前星星的y都是小于这个星的，所以只用考虑x了，就将问题转化成了一维的。
问题转化成了：当前处理的星星的坐标为 $(x_i,y_i)$ 求之前处理过的星星中 $x\le x_i$ 的个数。树状数组可以解决。
（如果没有输入这个条件，就需要先排一下序）

代码

/*
注意：x的值可能是0,所以需要把所有的x都+1才能用树状数组
*/
#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
const int INF=10000000;
const int N=32002;
int level[15005];
int C[32005];
void Add(int x,int d)
{
    while(x<=N)
    {
        C[x]+=d;
        x+=x&(-x);
    }
}
int Sum(int x)
{
    int ans=0;
    while(x>0)
    {
        ans+=C[x];
        x-=x&(-x);
    }
    return ans;
}
int main()
{
    int  n;
    int x,y;
    memset(C,0,sizeof(C));
    memset(level,0,sizeof(level));
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
          scanf("%d%d",&x,&y);
          level[Sum(x+1)]++;
          Add(x+1,1);
    }
    for(int i=0;i<n;i++)
    {
        printf("%d\n",level[i]);
    }
}