python经典逆序对问题_AcWing 788. 逆序对的数量（Python版本）

题目描述

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

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

输入格式

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

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

输出格式

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

数据范围

1≤n≤1000000

样例

输入样例：

6

2 3 4 5 6 1

输出样例：

5

算法1

(归并排序) $O(n\log n)$

Python 代码

"""

求逆序对的数目

利用归并排序的思想

"""

def merge_sort(nums):

if len(nums) <= 1:

return 0

mid = len(nums) // 2

L = nums[:mid]

R = nums[mid:]

ans = merge_sort(L) + merge_sort(R)

# 归并的过程

i = j = k = 0

while i < len(L) and j < len(R):

if L[i] <= R[j]:

nums[k] = L[i]

k += 1

i += 1

else:

nums[k] = R[j]

ans += len(L) - i

k += 1

j += 1

while i < len(L):

nums[k] = L[i]

k += 1

i += 1

while j < len(R):

nums[k] = R[j]

k += 1

j += 1

return ans

if __name__ == "__main__":

n = int(input())

nums = list(map(int, input().split()))

print(merge_sort(nums))