给定一个长度为 n 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i < j 且 a[i] > a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1 ≤ n ≤ 100000,
数列中的元素的取值范围 [1, 109]。
输入样例:
6
2 3 4 5 6 1
输出样例:
5
代码:
通过用归并排序计算逆序对:
#include <iostream>
using namespace std;
const int N = 100010;
int a[N], temp[N];
long long merge_sort(int l, int r)
{
if ( l >= r) return 0;
int mid = l + r >> 1;
long long res = merge_sort(l, mid) + merge_sort(mid + 1, r);
int i = l, j = mid + 1, k = 0;
while (i <= mid && j <= r)
{
if(a[i] <= a[j]) temp[k ++] = a[i ++];
else
{
temp[k ++] = a[j ++];
res += mid - i + 1;
}
}
while (i <= mid) temp[k ++] = a[i ++];
while (j <= r) temp[k ++] = a[j ++];
for (int i = l, j = 0; i <= r; i ++, j ++) a[i] = temp[j];
return res;
}
int main()
{
int n; cin >> n;
for (int i = 0; i < n; i ++ ) cin >> a[i];
cout << merge_sort(0, n - 1);
}