35.数组中的逆序对
题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007。
示例1
输入
1,2,3,4,5,6,7,0
输出
7
记录
归并排序
分治思想,步骤:
分解: 待排序的区间为 [l, r],找出中间元素,我们把 [l, r]分成 [l, m]和 [m + 1, r];
解决: 使用归并排序递归地排序两个子序列;
合并: 将两个排好序的子序列 [l, m] 和 [m + 1, r]合并。
每次将左右两边序列的最大元素进行比较,把其中较大的元素放到最终数组的末尾,然后在依次和序列中第二大的元素进行比较;
如果左序列的最大元素较大:说明该元素比右序列中现存的所有的元素都要大,则可以和右序列中所有的元素构成逆序对,即**(j+1)**个逆序对;
左序列未走完,将左序列剩下的元素合并到最终数组中;
右序列未走完,将右序列剩下的元素合并到最终数组中;
# -*- coding:utf-8 -*-
class Solution:
def InversePairs(self, data):
# write code here
if len(data) > 1:
#分解
mid = len(data) / 2
left_half = data[:mid]
right_half = data[mid:]
left_count = self.InversePairs(left_half)
right_count = self.InversePairs(right_half)
#指针,指向末尾
i,j,k,count = len(left_half)-1,len(right_half)-1,len(data)-1,0
#合并
while i >= 0 and j >= 0:
if left_half[i] < right_half[j]:
data[k] = right_half[j]
j = j - 1
k = k - 1
else:
data[k] = left_half[i]
count += (j+1)
i = i - 1
k = k - 1
while i >= 0:
data[k] = left_half[i]
k = k - 1
i = i - 1
while j>=0:
data[k] = right_half[j]
k = k - 1
j = j - 1
return (count + left_count + right_count)%1000000007
else:
return 0