AcWing 788. 逆序对的数量

题目

给定一个长度为n的整数数列，请你计算数列中的逆序对的数量。

逆序对的定义如下：对于数列的第 i 个和第 j 个元素，如果满足 i < j 且 a[i] > a[j]，则其为一个逆序对；否则不是。

输入格式

第一行包含整数n，表示数列的长度。

第二行包含 n 个整数，表示整个数列。

输出格式

输出一个整数，表示逆序对的个数。

数据范围

1 <= n <= 100000

输入样例：

6
2 3 4 5 6 1

输出样例：

5

题解

利用归并排序的时候,两段区间已经分别排好序,那么在合并的时候,当有两个数字时逆序对时,可以发现,大的那个数字的后面那个区间的数字都是大于小的那个数的,所以直接加上 mid - i + 1j即可,所以写一个归并排序,在排序的时候顺便把数量计算了

int n,a[MAXN],tmp[MAXN]; 
ll ans;

//求区间里的逆序对数量 
ll merge_sort(int l,int r){
	if(l >= r)return 0;
	int mid = l + r >> 1;
	ll res = merge_sort(l,mid) + merge_sort(mid+1,r);
	
	int k=0,i=l,j=mid+1;
	while(i <= mid && j <= r){
		if(a[i] <= a[j]){   //把小的放进临时数组中 
			tmp[k++] = a[i++];
		}
		else{
			tmp[k++] = a[j++];
			res += mid - i + 1;
		}
	}
	
	while(i <= mid)tmp[k++] = a[i++];
	while(j <= r)tmp[k++] = a[j++];
	
	for(int i=l,j=0;i<=r;i++){
		a[i] = tmp[j++];
	}
	return res; 
}

//归并排序 
int main(){
	cin >> n;
	ans = 0;
	for(int i=0;i<n;i++){
		cin >> a[i];
	}
	ans = merge_sort(0,n-1) ;
	cout << ans << endl;
	
	return 0;
}