leetcode 剑指offer51:数组中的逆序对 归并排序

题目描述:
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。

示例:
输入: [7,5,6,4]
输出: 5

限制:0 <= 数组长度 <= 50000

思路1: 暴力求解 遍历两趟 但是会超时,看到那个限制范围就可以推断出来

class Solution:
    def reversePairs(self, nums: List[int]) -> int:
        # 超时
        count = 0
        for i in range(len(nums)-1):
            for j in range(i+1, len(nums)):
                if nums[i] > nums[j]:
                    count += 1
        return count

思路2:
采用归并排序,归并排序的思想就是逆序的调换过来,这样在调换的过程就可以统计逆序对的数量;主要参考的题解链接
在这里插入图片描述
「归并排序」与「逆序对」是息息相关的。归并排序体现了 “分而治之” 的算法思想,具体为:
分: 不断将数组从中点位置划分开(即二分法),将整个数组的排序问题转化为子数组的排序问题;
治: 划分到子数组长度为 1 时,开始向上合并,不断将 较短排序数组 合并为 较长排序数组,直至合并至原数组时完成排序;
合并阶段 本质上是 合并两个排序数组 的过程,而每当遇到 左子数组当前元素 > 右子数组当前元素 时,意味着 「左子数组当前元素 至 末尾元素」 与 「右子数组当前元素」 构成了若干 「逆序对」 。 用两个指针完成。

算法流程:
merge_sort() 归并排序与逆序对统计:
终止条件: 当 l≥r 时,代表子数组长度为 1 ,此时终止划分;
递归划分: 计算数组中点 m ,递归划分左子数组 merge_sort(l, m) 和右子数组 merge_sort(m + 1, r) ;
合并与逆序对统计:
暂存数组 nums 闭区间 [i,r] 内的元素至辅助数组 tmp ;
循环合并: 设置双指针 i , j 分别指向左 / 右子数组的首元素;(注意下面四个情况的讨论顺序是必要的,不然将会出现数组越界)
当 i=m+1 时: 代表左子数组已合并完,因此添加右子数组当前元素 tmp[j] ,并执行j=j+1 ;
否则,当 j=r+1 时: 代表右子数组已合并完,因此添加左子数组当前元素 tmp[i] ,并执行i=i+1 ;
否则,当tmp[i]≤tmp[j] 时: 添加左子数组当前元素tmp[i] ,并执行 i=i+1;
否则(即 tmp[i]>tmp[j])时: 添加右子数组当前元素tmp[j] ,并执行 j=j+1 ;此时构成 m−i+1 个「逆序对」,统计添加至 res ;
返回值: 返回直至目前的逆序对总数 res ;
代码:

class Solution:
    def reversePairs(self, nums: List[int]) -> int:
        def merge_sort(l, r):
            if l >= r:
                return 0
            m = l + (r-l)//2
            res = merge_sort(l, m) + merge_sort(m+1, r)
            temp[l:r+1] = nums[l:r+1]
            i, j = l, m+1
            for k in range(l, r+1):  # 注意下面四个情况的讨论顺序是必要的,不然将会出现数组越界)
                if i == m+1:
                    nums[k] = temp[j]
                    j += 1
                elif j == r + 1:
                    nums[k] = temp[i]
                    i += 1
                elif temp[i] <= temp[j]:
                    nums[k] = temp[i]
                    i += 1
                elif temp[i] > temp[j]:
                    nums[k] = temp[j]
                    j += 1
                    res += m+1-i
            return res

        temp = [0] * len(nums)
        r = merge_sort(0, len(nums)-1)
        # print(nums)
        return r
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值