【剑指offer】逆序对

利用归并排序来解这道题。

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)

先将原序列排序,然后从排完序的数组中取出最小的,它在原数组中的位置表示有多少比它大的数在它前面,每取出一个在原数组中删除该元素,保证后面取出的元素在原数组中是最小的,这样其位置才能表示有多少比它大的数在它前面,即逆序对数。

发布了316 篇原创文章 · 获赞 96 · 访问量 11万+

猜你喜欢

转载自blog.csdn.net/weixin_31866177/article/details/99255071