利用归并排序来解这道题。
class Solution:
def InversePairs(self, data):
# 采用归并排序的思想,时间复杂度为O(nlogn)
global count
count = 0
def merge(array):
global count
if len(array) <= 1:
return array
k = len(array)//2
left = merge(array[:k])
right = merge(array[k:])
l = 0
r = 0
result = []
while l<len(left) and r<len(right):
if left[l]<=right[r]:
result.append(left[l])
l+=1
else:
result.append(right[r])
r += 1
count += len(left)+1
result += left[l:]
result += right[r:]
return result
merge(data)
return count%1000000007
if __name__ == '__main__':
data = [1,2,3,4,5,6,7,0]
print(Solution().InversePairs(data))
归并排序示意图如下:
还有一种,虽然只ac 50%,但是思想很简单
class Solution:
def InversePairs(self, data):
count = 0
copy = []
for i in data:
copy.append(i)
copy.sort()
for i in range(len(copy)):
count += data.index(copy[i])
data.remove(copy[i])
return count%1000000007
a=[1,2,3,4,5,6,7,0]
Solution().InversePairs(a)
先将原序列排序,然后从排完序的数组中取出最小的,它在原数组中的位置表示有多少比它大的数在它前面,每取出一个在原数组中删除该元素,保证后面取出的元素在原数组中是最小的,这样其位置才能表示有多少比它大的数在它前面,即逆序对数。