acwing——788. 逆序对的数量

788. 逆序对的数量

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

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

输入格式

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

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

输出格式

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

数据范围

1≤n≤1000001≤n≤100000

输入样例：

6
2 3 4 5 6 1

输出样例：

5

解题思路：

在分治后的每一层合并中顺便求出逆序对数量是这个题想法的由来，归并排序分治我们求的是从小到大的顺序，我们所求的逆序对恰好是逆序数量。可以边归并边求。

分三种情况：

第三种情况：当 q[i] > q[j] 时,由于i - mid 之间的所有数都已排好序且均大于 q[i]，所以 i - mid 之间的所有数也必然大于 q[j].
因此从 i - j 之间的逆序对数量便为 mid - i + 1.

还要注意：数据量大，用long long

程序代码：

 

#include<iostream>
using namespace std;

const int N = 100010;
int n;
int q[N], tmp[N];


long long merge_sort(int l, int r)
{
    if (l>=r) return 0;
    int mid = l + r >> 1;
    long long res = merge_sort(l,mid) + merge_sort(mid+1,r);
    
    //归并
    int k = 0, i = l, j = mid + 1;
    while (i<=mid && j<=r)
    {
        if (q[i]<=q[j]) tmp[k++] = q[i++];
        else
        {
            tmp[k++] = q[j++];
            res += mid - i + 1;
        }
    }
    while (i<=mid) tmp[k++] = q[i++];
    while (j<=r) tmp[k++] = q[j++];
    
    for (int i=l,j=0;i<=r;i++,j++) q[i] = tmp[j];
    
    return res;
}


int main()
{
    cin>>n;
    for (int i=0;i<n;i++) cin>>q[i];
    cout<<merge_sort(0,n-1)<<endl;
    return 0;
}