树状数组求解逆序对

逆序对

题目描述

猫猫 TOM 和小老鼠 JERRY 最近又较量上了，但是毕竟都是成年人，他们已经不喜欢再玩那种你追我赶的游戏，现在他们喜欢玩统计。

最近，TOM 老猫查阅到一个人类称之为“逆序对”的东西，这东西是这样定义的：对于给定的一段正整数序列，逆序对就是序列中 a[i]>a[j]且i<j的有序对。知道这概念后，他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。注意序列中可能有重复数字。

Update:数据已加强。

输入格式

第一行，一个数 n ，表示序列中有 n 个数。

第二行 n 个数，表示给定的序列。序列中每个数字不超过 10^9。

输出格式

输出序列中逆序对的数目。

输入输出样例

输入 #1复制

6
5 4 2 6 3 1

输出 #1复制

11

说明/提示

对于 25%25% 的数据，n≤2500

对于 50%50% 的数据，n≤4×10⁴

对于所有数据，n≤5×10⁵

请使用较快的输入输出

应该不会 O(n²)过 50 万吧 by chen_zhe

---

题解

思路

第一步：离散化处理数据，确保数组的空间大小在50w以内；
第二步：利用树状数组求逆序对个数。

实现

先定义全局变量

int n;//数组大小
int b[500002];//离散化处理后的数组
int tree[500002];//用于求逆序对的树状数组
struct point{
	int index,value;
}a[500002];

离散化处理

例如：
对数组 400 410 601 711 321 611进行离散化处理
得到b数组为 2 3 4 6 1 5

所以代码如下：

bool cmp(point q,point w){
	if(q.value==w.value){
		return (q.index<w.index);
	}else {
		return (q.value<w.value);
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i].value;
		a[i].index=i;
	}
	sort(a+1,a+n+1,cmp);
	for(int i=1;i<=n;i++){
		b[a[i].index]=i;
	}
	……
}

树状数组求解
如图：
这是一个经典的BIT，由15个结点组成，编号为1~15。
灰色结点为BIT中的结点（虚线稍后叙述），每一层结点的lowbit相同，而且lowbit越大，越靠近结点。虚线为BIT的边。注意index=0的结点为虚拟结点，并不是树的一部分，但能让算法更易理解。对于每一个灰色结点i，都有属于一个以自己为结尾的水平长条（对于lowbit==1的i结点，该长条就是它本身），该长条表示对应区间内元素的和。

int lowbit(int p){
	return p&-p;
}//p的二进制表达式中最右边的1所对应的值

对于结点i，若它是左结点，那么父结点为i+lowbit(i)，若它是右结点，那么父结点为i-lowbit(i)。

单点更新

void add(int p,int d){
	while(p<=n){
		tree[p]+=d;
		p+=lowbit(p);
	}
}

如图，index=3的元素改变时，需要将index=4和index=8元素所对应的长条的值改变，则要两次使用i=i+lowbit(i)。

区间查询

int query(int p){
	int sum=0;
	while(p>0){
		sum+=tree[p];
		p-=lowbit(p);
	}
	return sum;
}

如图，求index=11的之前所有元素的和，则需要两次调用i=i-lowbit(i)，分别把index=11，index=10和index=8的对应长条的值相加即可。

最后，考虑如何用树状数组求解问题
考虑第i个数,让tree[b[i]]=1(tree数组初始时元素均为0)，然后计算前i-1个数中有没有比b[i]大的，query(b[i]-1)的值就表示tree数组中前b[i]-1个数中出现的数的个数（<=i），这些树是在考虑第i个数之前就考虑过的，即这些数的下标小于i，且这些数比b[i]小，故i-1-query(b[i]-1)就是关于第i个数与前面所有数个逆序对个数。
所以，代码如下：

int main (){
	long long ans=0;
	for(int i=1;i<=n;i++){
		ans+=i-1-query(b[i]-1);
		add(b[i],1);
	}
	cout<<ans;
	return 0;
}

题目来源：洛谷P1908
参考资料：《算法竞赛入门经典——训练指南》