第一次接触树状数 很神奇的一个结构,适用于快速修改数组和数组前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;
}
}
}