在归并排序的基础上,增加一行代码即可
#include<iostream>
#define N 100000
long long res = 0;
void __merge(int arr[], int left, int mid, int right)
{
int *aux = new int[right - left + 1];
// 拷贝一遍数组
for (int i = left; i <= right; i++)
aux[i - left] = arr[i];
int i = left, j = mid + 1;
// arr数组用来排序,aux数组是原来的数组序列
for (int k = left; k <= right; k++) {
if (i > mid) { arr[k] = aux[j - left]; j++; } // 左边已排序完成
else if (j > right) { arr[k] = aux[i - left]; i++; } // 右边已排序完成
else if (aux[i - left] < aux[j - left]) { arr[k] = aux[i - left]; i++; } // 左边小于右边
else { // 右边小于左边
arr[k] = aux[j - left];
res += mid + 1 - i;
j++;
}
}
delete[]aux;
}
// 递归使用归并排序,对arr[l...r]的范围进行排序
void __mergeSort(int arr[], int left, int right) {
if (left >= right)
return;
int mid = left + (right - left) / 2;
__mergeSort(arr, left, mid);
__mergeSort(arr, mid + 1, right);
__merge(arr, left, mid, right);
}
int main()
{
int n;
int arr[N];
std::cin >> n;
for (int i = 0; i < n; i++)
std::cin >> arr[i];
__mergeSort(arr, 0, n - 1);
std::cout << res << std::endl;
}