数据结构实验之排序五：归并求逆序数

数据结构实验之排序五：归并求逆序数

Description

对于数列a1,a2,a3…中的任意两个数ai,aj (i < j)，如果ai > aj,那么我们就说这两个数构成了一个逆序对；在一个数列中逆序对的总数称之为逆序数，如数列 1 6 3 7 2 4 9中，(6,4)是一个逆序对，同样还有(3,2),(7,4),(6,2)，(6,3)等等，你的任务是对给定的数列求出数列的逆序数。

Input

输入数据N(N <= 100000)表示数列中元素的个数，随后输入N个正整数，数字间以空格间隔。

 

Output

输出逆序数。

Sample

Input 

10
10 9 8 7 6 5 4 3 2 1

Output 

45

#include <bits/stdc++.h>
using namespace std;
long long ans;
int a[100005];
void merg(int s1, int m1, int s2, int m2)
{
    int q = 0;
    int tem[100005];
    int q1 = s1;
    int q2 = s2;
    while(q1<=m1&&q2<=m2)
    {
        if(a[q1]<=a[q2])
        {
            tem[q++] = a[q1++];
        }
        else
        {
            tem[q++] = a[q2];
            ans+=m1-q1+1;
            q2++;
        }
    }
    while(q1<=m1)
    {
        tem[q++] = a[q1++];
    }
    while(q2<=m2)
    {
        tem[q++] = a[q2++];
    }
    for(int i = s1; i<=m2;i++)
        a[i] = tem[i-s1];

}
void merge_sort(int l, int r)
{
    if(l<r)
    {
        int mid = (l+r)/2;
        merge_sort(l, mid);
        merge_sort(mid+1, r);
        merg(l, mid, mid+1, r);
    }
}
int main()
{
    int n, i;
    ans = 0;
    scanf("%d", &n);
    for(i = 0;i<n;i++)
    {
        scanf("%d", &a[i]);
    }
    merge_sort(0, n-1);
    printf("%lld\n", ans);
    return 0;
}