新人第一次接触树状数组,上课也没有好好听讲,这里把做题的时候的一些经验稍微总结一下以免以后忘掉:
1、对于一个树状数组c[n],其中每个元素的值是对应的原数组a[n - 2^k +1]到a[n]的所有数字的和;
2、其中的k为数字n的最低的不为0的位的位置,习惯上将2^k称为low_bit,求low_bit的方法为n&(-n);
3、每一个节点c[i]管辖的范围的宽度都是2的幂数,low_bit(i)求出来的就是把i用2的次幂数进行分解求和的时候指数最小的那一项,c[i+low_bit]得到的就是c[i]的父节点,c[i]的值是其父节点管辖的值的一部分,因此要更新第i个值的时候,要把c[i]的所有父节点依次更新,以下是简单的函数实现:
void sum_update(int i,int val){
while(i<=maxn){
c[i]+=val;
i += low_bit(i);
}
}
4、求数组的前i项和的时候,c[i-low_bit(i)]即为与c[i]管辖区间不重叠的前一个区间的各个数的和,因此只要不断减这一个值,最终就可以求前n项和,当求某一个区间的和的时候,只需要两个前n项和相减即可,以下是简单的代码实现:
int getsum(int i){
int nSum = 0;
while(i>0){
nSum += c[i];
i -= low_bit(i);
}
return nSum;
}
5、值得注意的一点就是树状数组不能有标号为0的元素,因为low_bit(0)==0,所以会陷入死循环,在建立树状数组的时候可以对全部的元素编号加一,处理的时候要格外小心。
6、言归正传,关于stars这道题最巧妙的地方在于利用星星坐标的有序性,完全不需要使用y坐标,直接用x坐标对数组进行多次更新,并且在更新后求和,相当于一个滚动数组吧,有时间还会多做几道关于树状数组的题目。
7、说的不是很清楚,主要是给自己看的防止自己忘掉,请看官多多包涵~
#include<iostream>
#include<cstdio>
using namespace std;
#define maxn 32010
#define maxs 15010
int cnt[maxs]={0};
int tree[maxn]={0};
int N;
int lowbit(int n){
return n&(-n);
}
int getsum(int x){
int tsum = 0;
while(x>0){
tsum += tree[x];
x -= lowbit(x);
}
return tsum;
}
void add(int x){
while(x<maxn){
tree[x]++;
x += lowbit(x);
}
return;
}
int main(){
int x,y;
scanf("%d",&N);
for(int i=0;i<N;++i){
scanf("%d%d",&x,&y);
int sum = getsum(x+1);
cnt[sum]++;
add(x+1);
}
for(int i=0;i<N;++i){
printf("%d\n",cnt[i]);
}
//system("pause");
return 0;
}