剑指 Offer 51题:数组中的逆序对

1.题目描述

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

示例1:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof/

2.思路分析

「归并排序」与「逆序对」是息息相关的。归并排序体现了 “分而治之” 的算法思想,具体为:

  • : 不断将数组从中点位置划分开(即二分法),将整个数组的排序问题转化为子数组的排序问题;
  • : 划分到子数组长度为 1 时,开始向上合并,不断将 较短排序数组 合并为 较长排序数组,直至合并至原数组时完成排序;

如下图所示,为数组 [7, 3, 2, 6, 0, 1, 5, 4] 的归并排序过程。

合并阶段 本质上是 合并两个排序数组 的过程,而每当遇到 左子数组当前元素 > 右子数组当前元素 时,意味着 「左子数组当前元素 至 末尾元素」 与 「右子数组当前元素」 构成了若干 「逆序对」 。

因此,考虑在归并排序的合并阶段统计「逆序对」数量,完成归并排序时,也随之完成所有逆序对的统计。

实际上,只需要在归并排序的代码中加一行程序即可解决本题 

如下图所示,为左子数组 [2, 3, 6, 7] 与 右子数组 [0, 1, 4, 5] 的合并与逆序对统计过程。(部分) 

2.1 初始化 

2.2 依次比较并进行归并排序和逆序对的统计

 后面的步骤同理,此处省略......

3.解答 

class Solution {
    //归并排序,求逆序对
    
    //统计逆序对的数量
    int count = 0;

    public int reversePairs(int[] nums) {
        //归并排序的同时统计逆序对的数量
        mergeSoer(nums, 0, nums.length - 1);

        return count;
    }

    //辅助方法:归并排序
    public void mergeSoer(int[] arr, int low, int high) {
        int middle = (high+low)/2;
        if(low < high) {
            // 处理左边
            mergeSoer(arr, low, middle);
            // 处理右边
            mergeSoer(arr, middle+1, high);
            // 归并
            merge(arr, low, middle, high);
        }
    }

    //辅助方法:合并区间
    public void merge(int[] arr, int low, int middle, int high) {
        // 用于存储归并后的临时数组
        int[] temp = new int[high-low+1];
        // 记录第一个数组中需要遍历的下标
        int i = low;
        // 记录第二个数组中需要遍历的下标
        int j = middle + 1;
        // 记录临时数组中的下标
        int index = 0;

        while(i <= middle && j <= high) {
            if(arr[i] <= arr[j]) {
                temp[index] = arr[i];
                i++;
            }else {
                //统计逆序对的数量
                count = count + (middle - i + 1);
                temp[index] = arr[j];
                j++;
            }
            index++;
        }

        // 处理剩下的数据
        while(i <= middle) {
            temp[index] = arr[i];
            i++;
            index++;
        }
        while(j <= high) {
            temp[index] = arr[j];
            j++;
            index++;
        }

        // 把临时数组temp中的元素返回arr中
        for(int k = 0; k < temp.length; k++) {
            arr[k+low] = temp[k];
        }
    }
}

时间复杂度 O(N log N) : 其中 N 为数组长度;归并排序使用 O(N log N) 时间;
空间复杂度 O(N) : 辅助数组 tmp 占用 O(N)大小的额外空间;

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值