思路在代码注释中
给定一个长度为 n 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i<j 且 a[i]>a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤100000,
数列中的元素的取值范围 [1,109]。
输入样例:
6
2 3 4 5 6 1
输出样例:
5
/**
* 先回顾归并排序的思想:
* 1. [L,R] => [L, mid], [mid + 1, R]
* 2. 归并排序[L, mid], [mid + 1, R]
* 3. 归并,将左右两个有序序列合并成一个有序序列
*
* 先根据分治的思想分为:
* 1.所有逆序对都在左半边 (L, mid)
* 2.所有逆序对都在右半边 (mid + 1, R)
* 3. 一个数在左半边,一个数在右半边
*
* 三者之和就是逆序对数量
*
* 我们可以发现,当左右两边都排好序的时候,第三步计数更简单
* 也就是归并排序的时候顺便记一下数,时间复杂度很低
*
**/
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
int n;
int a[N];
LL res = 0;
void merge_sort(int l, int r){
if (l >= r) return;
int mid = (l + r) / 2;
int i = l, j = mid + 1, tem[r - l + 1], k = 0;
merge_sort(l, mid);
merge_sort(mid + 1, r);
while (i <= mid && j <= r){
if (a[i] <= a[j]){
tem[k++] = a[i++];
}
else{
tem[k++] = a[j++];
res += (mid - i + 1);
}
}
while (i <= mid) tem[k++] = a[i++];
while (j <= r) tem[k++] = a[j++];
for (int i = l, j = 0; j < k; i++, j++){
a[i] = tem[j];
}
}
int main(){
cin >> n;
for (int i = 0; i < n; i++){
cin >> a[i];
}
merge_sort(0, n - 1);
cout << res;
return 0;
}