题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007。例如:在数组{7,5,6,4}中,一共存在5个逆序对,分别是(7,6),(7,5),(7,4),(6,4),(5,4)
思路
类似归并排序。
- 在递归拆分数据时,调整相邻数字位置,算出逆序对的个数
- 在左区间和右区间合并时,如果左边i大于右边j,则直接返回逆序对个数mid-i+1(因为左边区间是排好序的,最小的都比j大,i后面的也一定大于j,就不用再算了)
- 如果左边已经走完了,右边直接接在后面。同理,左边直接接在后面
# -*- coding:utf-8 -*-
class Solution:
def InversePairs(self, data):
# write code here
if not data:
return 0
return self.merge_sort(data[:],data[:],0,len(data)-1)%1000000007
def merge_sort(self,temp,data,left,right):
if right-left<1:
return 0
if right-left==1:
if data[left]<=data[right]:
return 0
else:
temp[left],temp[right]=data[right],data[left]
return 1
mid=(left+right)//2
#这里temp和data互换,是因为让temp保存上一步的结果,data为这次正在运行后的结果
res=self.merge_sort(data,temp,left,mid)+self.merge_sort(data,temp,mid+1,right)
i=left
j=mid+1
index=left
while i<=mid and j<=right:
if data[i]<=data[j]:
temp[index]=data[i]
i+=1
index+=1
else:
res+=mid-i+1
temp[index]=data[j]
j+=1
index+=1
while i<=mid:
temp[index]=data[i]
i+=1
index+=1
while j<=right:
temp[index]=data[j]
j+=1
index+=1
return res
测试用例
if __name__=='__main__':
s=Solution()
numbers1=[7,5,6,4]
numbers2=[3,4,5,1,2]
print(s.InversePairs(numbers1))
print(s.InversePairs(numbers2))
ps:这题想通过牛客网AC,需将上述代码多提交几次或者在网速好的地方提交