此题来源于剑指offer
题目描述:
在数组中的两个数字如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
从直观上来看,可以使用一个 O ( N 2 ) O(N^2) O(N2)的算法:顺序扫描整个数组,每扫描到一个数字时,逐个比较它与它后面数字的大小,从而获得逆序对数。
剑指offer上介绍了一种
O
(
N
l
o
g
N
)
O(NlogN)
O(NlogN)的方法,采用归并排序的思想,在归并操作中比较左右数字的大小时,可以获得逆序对的数目。
直接上代码
cnt = 0
def InversePairs(nums):
merge_sort(nums)
def merge_sort(nums):
length = len(nums)
if length<=1:
return nums
mid = length // 2
left = merge_sort(nums[:mid])
right = merge_sort(nums[mid:])
return merge(left,right)
def merge(left,right):
global cnt
res,i,j = [],0,0
while i < len(left) and j < len(right):
if left[i] <= right[j]:
res.append(left[i])
i += 1
else:
res.append(right[j])
j += 1
cnt += len(left) - i
# 处理剩余的数组
res += left[i:]
res += right[j:]
return res