树状数组--求比某一位置x小的数的个数
程序员文章站
2022-06-15 11:05:21
...
好好看了一些树状数组的东西,看了一些原理,以1,2,3,4,5为例,代码如下:
#include<iostream>
#include<cstring>
using namespace std;
const int MAXX=10000;
int a[MAXX];//原始数组
int c[MAXX]; //树状数组
int S_L[MAXX];//记录每个位置左边小于他的个数
int S_R[MAXX];//记录每个位置右边小于他的个数
int max_a; //最大值
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int v)
{
for(; x<=max_a; x+=lowbit(x))
c[x]+=v;
}
int sum(int x)
{
int sum=0;
for(; x>0; x-=lowbit(x))
{
sum+=c[x];
}
return sum;
}
int main()
{
int n;
max_a=0;
cin>>n;
for(int i=1; i<=n; i++)
{
cin>>a[i];
max_a=max(max_a,a[i]);
}
for(int i=1; i<=n; i++)
{
S_L[i]=sum(a[i]);
add(a[i],1);
}
memset(c,0,sizeof(c)); //把数反过来 就是右边小的数的个数
for(int i=n; i>=1; i--)
{
S_R[i]=sum(a[i]);
add(a[i],1);
}
cout<<"the number of minner at left: "<<endl;
for(int i=1; i<=n; i++)
cout<<S_L[i]<<" ";
cout<<endl;
cout<<"the number of minner at right: "<<endl;
for(int i=1; i<=n; i++)
cout<<S_R[i]<<" ";
cout<<endl;
}
推荐阅读