牛客网剑指offer 数组中的逆序对 Python3

最新推荐文章于 2020-06-03 18:18:48 发布

原创最新推荐文章于 2020-06-03 18:18:48 发布 · 210 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 同时被 2 个专栏收录

9 篇文章

订阅专栏

剑指offer

2 篇文章

订阅专栏

本文介绍了一种求解逆序对数量的高效算法，并提供了Python实现代码。该算法通过分治策略，利用归并排序的思想，计算数组中逆序对的总数。逆序对是指在数组中，对于任意两个元素(i,j)，如果i<j且data[i]>data[j]，则称(i,j)构成逆序对。算法首先将原数组复制一份，然后递归地将数组分为两部分进行排序，同时统计逆序对数量。最后，将排序后的两部分合并，继续统计逆序对数量。

思路参考了另一篇博客，此处只是将原作者的代码翻译为Python版本：https://blog.youkuaiyun.com/lym940928/article/details/91354887

# -*- coding:utf-8 -*-
class Solution:
    def InversePairs(self, data):
        length=len(data)
        if length<=0:
            return 0
        copy=[]
        for i in range(length):
            copy.append(data[i])
        count=self.InverseHelper(data,copy,0,length-1)
        return count%1000000007
    
    def InverseHelper(self, data, copy, start, end):
        if start==end:
            copy[start]=data[start]
            return 0
        length=(end-start)/2
        left=self.InverseHelper(copy,data,start,start+length)
        right=self.InverseHelper(copy,data,start+length+1,end)
        
        i=start+length
        j=end
        indexcopy=end
        count=0
        while i>=start and j>=start+length+1:
            if data[i]>data[j]:
                copy[indexcopy]=data[i]
                indexcopy-=1
                i-=1
                count=count+j-start-length
            else:
                copy[indexcopy]=data[j]
                indexcopy-=1
                j-=1
        while i>=start:
            copy[indexcopy]=data[i]
            indexcopy-=1
            i-=1
        while j>=start+length+1:
            copy[indexcopy]=data[j]
            indexcopy-=1
            j-=1
        return left+right+count