788. 逆序对的数量

AcWing 788. 逆序对的数量

题目：

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

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

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

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

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

数据范围
1≤n≤100000
输入样例：
6
2 3 4 5 6 1
输出样例：
5

代码：

#include <cstdio>
#include <iostream>

using namespace std;

const int N = 100005;
typedef long long LL;
//注意当给出的数据序列是降序序列时，逆序对数为n-1 + n-2 + .. + 1 约等于n²/2
//又因为n最大为1e5,则结果可能为5e9，大于int范围1e9,故用long long

LL num[N], t[N];

LL mergeSort(int l, int r)
{
    if (l >= r)return 0;
    LL ans = 0;
    int mid = l + r >> 1;
    ans += mergeSort(l, mid);//统计逆序对中两个数都在左区间的逆序对数
    ans += mergeSort(mid+1, r);//统计逆序对中两个数都在右区间的逆序对数

    int  i = l, j = mid+1, k = 0;
    while (i <= mid && j <= r)
        if (num[i] <= num[j])//注意这里需要是<=,两数相等时不是逆序对
            t[k++] = num[i++];
        else
        {
            ans+=mid-i+1;//统计当逆序对中大的在左区间，小的在右区间时的逆序对数
            t[k++] = num[j++];
        }
    while (i <= mid) t[k++] = num[i++];
    while (j <= r) t[k++] = num[j++];

    for (i = 0; i < k; i++)
        num[l+i] = t[i];
    return ans;
}

int main()
{
    int n;
    cin>>n;
    for (int i = 0; i < n; i++)
        scanf("%d",&num[i]);

    cout<<mergeSort(0,n-1)<<endl;
    return 0;
}

题目解析