欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

树状数组--求比某一位置x小的数的个数

程序员文章站 2022-06-15 11:05:21
...

好好看了一些树状数组的东西,看了一些原理,树状数组--求比某一位置x小的数的个数以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;
}

相关标签: 树状数组