hdu 1541 树状数入门

第一次接触树状数  很神奇的一个结构,适用于快速修改数组和数组前n项求和!复杂度log(n);

题意:给出n个星星坐标   按y为第一关键字x为第二关键字的顺序给出,星星左下角所包含的星星的个数为该星星的等级,输出0~n-1等级星星的个数;

 这里给出图示方便了解树状数组结构:

主要掌握和理解几个函数的作用

lowbit(x)返回二进制表示最小进制幂2^k

sum(x)前x项和

add(x,y)x项修改增加y

代码:

#include<iostream>
#include<cstring>
using namespace std;
int N;
int ar[32005];              //数状数组
int c[32005];               //等级

int lowbit(int t)       //位运算,求最小幂2^k的k
{
    return t&(-t);
}
void add(int t,int v)           //对元素进行加法操作
{
    for(int i=t;i<=32005;i+=lowbit(i))
    {
        ar[i]+=v;
    }
}
int sum(int t)
{
    int s=0;
    for(int i=t;i>0;i-=lowbit(i))
    {
        s+=ar[i];
    }
    return s;
}
int main()
{
    int x,y;
    while(cin>>N)
    {
        memset(ar,0,sizeof(ar));
        memset(c,0,sizeof(c));
        for(int i=0;i<N;i++)
        {
            cin>>x>>y;
            c[sum(x+1)]++;          //等级为(sum(x+1)//求前面总星数)=前(x+1)项和
            add(x+1,1);             //第(x+1)个数组+1    数组下标从1开始  所以+1;
        }
        for(int j=0;j<N;j++)
        {
            cout<<c[j]<<endl;
        }
    }
}


 

原文地址:https://www.cnblogs.com/amourjun/p/5134194.html