算法笔记求序列A每个元素左边比它小的数的个数(树状数组和离散化)

 

 

 

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std ;

const int N = 100010 ;

struct node{
    int val,pos ;
}tmp[N];
int a[N] ;//离散化后的原始数组 
int c[N] ;//树状数组 


bool cmp(node st1,node st2){
    return st1.val < st2.val ;
}

int lowbit(int x){//取得最右边的一个1 
    return x&(-x) ;
}

int getSum(int x){//区间1~x的和 log(N) 
    int sum = 0 ;
    for(int i=x;i>0;i-=lowbit(i)){
        sum += c[i] ;
    }
    return sum ;
}

void update(int x,int v){//单点更新 log(N)
    for(int i=x;i<=N;i+=lowbit(i)){
        c[i] += v ;
    }
}


int main(){
    int n, x ;
    cin >> n ;
    
    memset(c,0,sizeof c) ;
    
    for(int i=0;i<n;i++){
        cin >> tmp[i].val ;
        tmp[i].pos = i ;
    }
    
    sort(tmp,tmp+n,cmp) ;
    
    for(int i=0;i<n;i++){
        if(i==0 || tmp[i-1].val != tmp[i].val)
            a[tmp[i].pos] = i + 1 ;//离散化 
        else{
            a[tmp[i].pos] = a[tmp[i-1].pos] ;
        }
    } 
    
    for(int i=0;i<n;i++){
        update(a[i],1) ;
        cout << getSum(a[i]-1) << endl ;
    }
    
    return 0 ;
    
}