【leetcode系列】【面试题】【困难】数组中的逆序对

题目:

题目链接: https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof/

解题思路:

自己的方法:

对数组进行遍历,添加一个数组保存已经遍历过的数字。每次遍历的时候,使用二分查找,找到当前数字应该在保存记录数组中插入的最右边的位置(如果有重复数字,插入到重复数字的最右边)

此时就可以根据当前的idx值,和插入的位置,计算出之前有多少个数字比当前数字大,更新结果集

每次需要二分查找一次,时间复杂度为logN,需要进行N次二分查找

所以总的时间复杂度应该是O(NlogN)

官方题解:

https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof/solution/shu-zu-zhong-de-ni-xu-dui-by-leetcode-solution/

代码实现:

class Solution:
    def reversePairs(self, nums: List[int]) -> int:
        def find_pos(nums, target):
            left = 0
            right = len(nums)
            fix_num = 0
            while left <= right:
                mid = (left + right) // 2
                if nums[mid] > target:
                    right = mid - 1
                else:
                    left = mid + 1

            if target < nums[mid]:
                return mid

            return mid + 1

        new_nums = []
        res = 0
        for idx, curr_num in enumerate(nums):
            if 0 == len(new_nums) or curr_num >= new_nums[-1]:
                new_nums.append(curr_num)
                continue

            pos = find_pos(new_nums, curr_num)
            res += idx - pos
            new_nums.insert(pos, curr_num)

        return res
发布了138 篇原创文章 · 获赞 13 · 访问量 2445

猜你喜欢

转载自blog.csdn.net/songyuwen0808/article/details/105723041