求逆序对的数量

题目描述:

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

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

输入格式:

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

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

输出格式:

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

数据范围:

1≤n≤100000，
数列中的元素的取值范围 [1,1e9]。

输入样例:

6
2 3 4 5 6 1

输出样例：

5

思路:

本题使用分治法解决问题，用到了归并排序的思路。

假设merge函数能够同时对序列进行排序并返回该序列中逆序对的数量。

现取mid=(l+r)/2

把所有逆序对分成3类：

第一类：两个元素都在mid左边

1.逆序对的数量：merge(l,mid);

第二类：两个元素都在mid右边

2.逆序对的数量：merge(mid+1,r);

第三类：两个元素在mid左右两边

3.逆序对的数量：a[i]>a[j] ΣSj=mid-i+1

AC代码:

#include <iostream>

using namespace std;

typedef long long LL;
int const N = 100010;
int a[N], tmp[N];

LL merge(int l, int r)
{
    if (l >= r) return 0;

    int mid = l + r >> 1;
    LL ans = merge(l, mid) + merge(mid + 1, r);

    int k = 0, i = l, j = mid + 1;

    while (i <= mid && j <= r)
    {
        if (a[i] <= a[j]) tmp[k++] = a[i++];
        else {
            tmp[k++] = a[j++];
            ans += mid - i + 1;
        }
    }

    while (i <= mid) tmp[k++] = a[i++];
    while (j <= r) tmp[k++] = a[j++];

    for (int i = l, j = 0; i <= r; i++, j++) a[i] = tmp[j];

    return ans;
}

int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++) cin >> a[i];
    cout << merge(0, n - 1);

    return 0;
}