树状数组求逆序对

1 逆序对是什么？

是指在一个$a$序列中，满足$i<j$且$a_i>a_j$的$i,j$总数和。
例如：

5 4 2 6 3 1

逆序对数就是11，即：

5 4 5 2 5 3 5 1 4 2 4 3 4 1 2 1 6 3 6 1 3 1

2 如何求逆序对

我们知道，树状数组是可以支持单点修改以及区间查询的，那么我们要是将树状数组维护的数据改为 比$x$小的数在$x$后面有$y$个，不就完成了逆序对的任务吗？
别急，我们还有一个优化，当逆序对的数据范围十分大的时候，例如最大可以达到$10^9$时，我们显然不可能开一个$10^9$大小的数组来存储这个信息，我们考虑离散化。
用一个pos[i]数组来记录第i个数在从小到大排序后在第几位，我们就可以抽象大小，只注意数据的大小关系。
代码实现如下：

#include <iostream>
#include <cstdio>
#include <algorithm>
#define ll long long
#define maxn 500001
using namespace std;
ll n,a[maxn],pos[maxn],b[maxn],ans;
ll lowbit(ll x){
	return x&-x;
}
void add(ll x,ll y){
	while(x<=n){
		b[x]+=y;
		x+=lowbit(x);
	}
}
ll query(ll x){
	int res=0;
	while(x>0){
		res+=b[x];
		x-=lowbit(x);
	}
	return res;
}
bool cmp(int x,int y){
	return a[x]<a[y];
}
int main(){
	scanf("%lld",&n);
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
		pos[i]=i;
	}
	stable_sort(pos+1,pos+1+n,cmp);
	for(int i=1;i<=n;i++){
		a[pos[i]]=i;
	}
	for(int i=n;i>=1;i--){
		ans+=query(a[i]-1);
		add(a[i],1);
	}
	printf("%lld\n",ans);
	return 0;
}

3 结语

树状数组十分有用，可以处理最大值，区间和等等问题，我们在求逆序对的时候也不要忘记了树状数组的使用方法。再者，sort函数会交换两个相等的数的位置，但是stable_sort就不同，它严格保证数据的前后关系，我们在处理问题时要灵活处理。