树状数组求逆序对


利用‘桶’的思想
a[i]表示 i 的个数
那么c[i]就表示范围在[i-lowbit(i)+1,i]的数的个数
树状数组求和操作的sum(t)表示[1,t]之间的数的个数
在for循环新输入一个数的时候,逆序对的数量应该增加“比新加入的数大的数”的数量(前面插入的数的下标肯定比现在的小)
如已经插入了4 3 1,现在插入2,那么逆序对应该增加2
即ans+=i-sum( t );
( t 是新插入的数,sum( t )是1-t之间的数的数量)
(i-sum[ t ]就是比t大的数的数量)

code:

#include
#include
typedef long long ll;
const int inf=0x3f3f3f3f;
const int inn=0x80808080;
using namespace std;
const int maxm=5e5+5;
int c[maxm];
int n;
ll ans;
int lowbit(int x){
    return x&(-x);
}
void add(int i,int t){
    while(i<maxm){//这里和maxm相关,而不是n,因为i是数据而不是a[i]是数据
        c[i]+=t;
        i+=lowbit(i);
    }
}
int sum(int i){
    int res=0;
    while(i){
        res+=c[i];
        i-=lowbit(i);
    }
    return res;
}
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        int t;
        cin>>t;
        add(t,1);
        ans+=i-sum(t);
    }
    for(int i=1;i<=n;i++){
        cout<<c[i]<<endl;
    }
    cout<<ans<<endl;
    return 0;
}

你可能感兴趣的:(树状数组)