高级数据结构第七章A . 数列(树状数组逆序对)

image

代码:

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=51000;
int a[maxn],n;
int tr[maxn],b[maxn],tr1[maxn];
int lowbit(int x){
	return x&-x;
}
void update(int tr[],int pos,int val){
	while(pos<maxn){
		tr[pos]+=val;
		pos+=lowbit(pos); 
	}
}
int query(int tr[],int pos){
	int res=0;
	while(pos){
		res+=tr[pos];
		pos-=lowbit(pos);
	}
	return res;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
    	scanf("%d",&a[i]);
    	a[i]++;
    	update(tr,a[i],1);
    	b[i]=query(tr,a[i]-1);
	}
	ll res=0;
	for(int i=n;i;i--){
		update(tr1,a[i],1);
		res+=b[i]*query(tr1,a[i]-1);
	}
	printf("%lld\n",res);
    return 0;
}
posted @ 2021-06-15 21:50  OvO1  阅读(42)  评论(0编辑  收藏  举报