归并排序应用-求逆序对数量

题目：

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

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

输入格式

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

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

输出格式

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

数据范围

1≤n≤1000001≤n≤100000

输入样例：

6
2 3 4 5 6 1

输出样例：

5

代码：

#include <iostream>
#include <algorithm>
using namespace std;

const int N = 1e7;

long arr[N];
long help[N];
int n;

long long ans;

void union_sort(int l, int r){
    if(l >= r)return;
    int mid = l + r >> 1;
    union_sort(l, mid);
    union_sort(mid+1,r);
    int i = l, j = mid + 1, k = 1;
    while(i <= mid && j <= r){
        if(arr[i] <= arr[j])
            help[k++] = arr[i++];
        else{
            help[k++] = arr[j++];
            ans += mid - i + 1;
        }
    }
    while(i <= mid)help[k++] = arr[i++];
    while(j <= r)help[k++] = arr[j++];
    for(int i = l, j = 1;i <= r;++i)
        arr[i] = help[j++];
}

int main(){
    cin >> n;
    for(int i = 1;i <= n;++i)cin >> arr[i];
    union_sort(1, n);
    cout << ans;
    return 0;
}