题目
给定一个长度为 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;
}