树状数组
树状数组可以快速地求取连续地字串和
给定一个n个元素的数组 A1、A2...An ,树状数组支持以下两种操作。
Add(x,d):让Ax增加d
Query(L,R):计算AL+AL+1...AR
【例题一】
LA4329pingpong
- 题目大意
给出n个数 a1、a2...an(1≤ai≤100000) ,设 ai 之前比 ai 小的数的个数为 Ci ,求 ∑ni=1Ci
- 分析
现在先考虑一个与此题无关的问题,给定一个数组,设计一种方法支持一种查询操作:查询这个数组中比x小的数的个数。这里x是可变的。一种做法是先排一次序,再遍历一次就可以了。复杂度是
O(nlgn)
,观察发现这里
ai
是个不大的数,为了节省时间,排序也可以不用了(或者叫计数排序)。建一个数组,统计出值为m的数的个数存入数组中下标为m的地方。这样也可以达到类似排序的效果。最后再遍历一次,累加数组中的值(到下标为x时停止)
再回到上面的问题,这里的区间是变化的,但仅仅是从前面的区间上增加了元素,所以还是可以用上个区间的计数结果进行更新就行。但对于每一个
ai
,在求
ai
之前的区间里比
ai
小的数的个数时都用去遍历累加一遍会太慢,而这里,树状数组就刚好是为了这种查询而设计的。
这个问题和求一组数的逆序对数几乎一样,只是把小改成大。
【例题二】
- 题目大意
n个星星,坐标 (x,y) ,设第i个星左下方(包括边界)星星的数量为z,则称level为x。求出所有level水平的数量。输入按照先y递增,y相同则x递增。
- 分析
初一看像是一个二维问题,但其实一维就可以解决。输入是先y递增,y相同则x递增。这个条件其实很重要。问题可以等价于求每个星左下方星星的数量。按照输入的顺序进行处理。可以发现对于当前要处理的星,之前星星的y都是小于这个星的,所以只用考虑x了,就将问题转化成了一维的。
问题转化成了:当前处理的星星的坐标为
(xi,yi)
求之前处理过的星星中
x≤xi
的个数。树状数组可以解决。
(如果没有输入这个条件,就需要先排一下序)
- 代码
/*
注意: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]);
}
}
【例题三】
树状数组用于优化
- 题目大意
a1、a2、...an 求这个序列中,严格递增的序列的个数。 (n≤100000,ai不超过int范围)
- 分析
直接dp是
O(n2)
的复杂度,会爆,考虑优化方法。
dp[i]
表示以i结尾的递增序列个数。
这里就可以用树状数组来求 ∑dp[j]