剑指offer数组中的逆序对

题目描述

在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007。例如：在数组{7,5,6,4}中，一共存在5个逆序对，分别是(7,6),(7,5),(7,4),(6,4),(5,4)

思路

     类似归并排序。

1. 在递归拆分数据时，调整相邻数字位置，算出逆序对的个数
2. 在左区间和右区间合并时，如果左边i大于右边j，则直接返回逆序对个数mid-i+1(因为左边区间是排好序的，最小的都比j大，i后面的也一定大于j，就不用再算了)
3. 如果左边已经走完了，右边直接接在后面。同理，左边直接接在后面

# -*- 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，需将上述代码多提交几次或者在网速好的地方提交