POJ 2352 Stars

来源:互联网 发布:大额借贷软件 编辑:程序博客网 时间:2024/06/08 13:19

新人第一次接触树状数组,上课也没有好好听讲,这里把做题的时候的一些经验稍微总结一下以免以后忘掉:

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 15010int 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;}
0 0
原创粉丝点击