逆序对的个数（归并排序C++）

题目

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

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

输入格式

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

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

输出格式

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

数据范围

1≤n≤100000，
数列中的元素的取值范围 [1,]。

输入样例：

6
2 3 4 5 6 1

输出样例：

5

 

题解 

 主要思路：

        在归并排序中，如果左半边的某个元素 a[i] 比右半边的元素 a[j] 大，那么 a[j] 是小于左半边剩余所有元素的，因此逆序对的个数就增加了 mid - i + 1 个。这是因为，左半边剩余的所有元素都可以和 a[j] 组成逆序对。由于左半边和右半边都是已经排好序的数组，因此我们可以保证统计到每一个逆序对，从而达到正确计数的目的。

#include<iostream>

using namespace std;

const int N = 1e6 + 10;

typedef long long LL;
int a[N], tmp[N];

LL msort(int l, int r) {
	if (l >= r)return 0;
	int mid = l + r >> 1;
	LL res = msort(l, mid) + msort(mid + 1, r);
	
	int i = l, j = mid + 1, k = 0;
	while (i <= mid && j <= r) {
		if (a[i] > a[j]) {
			tmp[k++] = a[j++];
			res += j - mid + 1;
		}
		else tmp[k++] = a[i++];
	}
	while(i <= mid)tmp[k++] = a[i++];
	while (j <= r)tmp[k++] = a[j++];

	for (i = l, k = 0; i <= r; i++, k++)
		a[i] = tmp[k];

	return res;
}

int  main() {
	int n;
	cin >> n;

	for (int i = 0; i < n; i++)
		scanf("%d", &a[i]);

	cout << msort(0, n - 1);

	return 0;
}