逆序对个数（归并排序）Python

逆序对个数(归并排序)

1. Description

给出一个数组a，问这个数组中有多少个逆序对。
逆序对定义: 若i < j且a[i] > a[j],则(ai, aj)是一个逆序对。
如数组3 4 1 2中的逆序对有(3,1),(3,2),(4,1),(4,2)，共4个逆序对

2. Output

第一行一个整数n，表示元素个数。
第二行n个空格分隔的整数ai。

3. Sample Input 1

5
3 3 4 1 2

4. Hint

50%的数据 1 <= 6000
100%的数据 1 <= 10的5次方；0 < ai < 2的31次方

5. Code

def merge_sort(s,l,r):
    if l >= r: return 0
    mid = l + r >> 1
    res = merge_sort(s,l,mid) + merge_sort(s,mid + 1,r)
    i = l
    j = mid + 1
    tmp = []
    while(i <= mid and j <= r):
        if(s[i] <= s[j]):
            tmp.append(s[i])
            i += 1
        else:
            res += (mid - i + 1)
            tmp.append(s[j])
            j += 1
            pass
        pass
    while(i <= mid):
        tmp.append(s[i])
        i += 1
        pass
    while(j <= r):
        tmp.append(s[j])
        j += 1
        pass
    s[l:r + 1] = tmp
    return res


n = eval(input())
s = list(map(int,input().split()))
print(merge_sort(s, 0, n - 1))