当前位置: 首页 > article >正文

数组中的逆序对(C++)

目录

1 问题描述

1.1 输入描述:

1.2 示例1

1.3 示例2

2 解题思路

2.1 暴力解法

2.2 归并排序法

3 代码实现

3.1 暴力解法

3.2 归并排序法

4 代码解析

4.1 暴力解法

4.1.1 初始化

4.1.2 判断是否是逆序对

4.2 归并排序法

4.2.1 InversePairs 主函数

4.2.2 归并排序 merge_sort__ 递归拆分

4.2.3 归并 merge__ 统计逆序对

5 总结


1 问题描述

在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P mod 1000000007
数据范围:  对于 50%50% 的数据, size≤104size≤104
对于 100%100% 的数据, size≤105size≤105

数组中所有数字的值满足 0≤val≤1090≤val≤109

要求:空间复杂度 O(n)O(n),时间复杂度 O(nlogn)O(nlogn)

1.1 输入描述:

题目保证输入的数组中没有的相同的数字

1.2 示例1

输入:

[1,2,3,4,5,6,7,0]

返回值:

7

1.3 示例2

输入:

[1,2,3]

返回值:

0

2 解题思路

2.1 暴力解法

通过两层嵌套循环,外层循环 i 遍历数组中的每个元素,内层循环 ji+1 开始,检查 nums[i] > nums[j] 是否成立,若成立则递增逆序对计数 p++。最终,返回 p % 1000000007 以避免结果溢出。

2.2 归并排序法

采用归并排序的方法高效计算数组中的逆序对。在 InversePairs 函数中,调用 merge_sort__ 递归地对数组进行归并排序,并在合并过程中计算逆序对。merge_sort__ 先递归地将数组拆分为左右两部分,再在 merge__ 过程中合并,同时统计逆序对。合并时,如果左半部分的 arr[i] > arr[j]j 来自右半部分),则意味着 arr[i] 及其后续所有元素均大于 arr[j],因此逆序对数 ret 需要累加 (mid - i + 1),并取模 1000000007 以防溢出。

3 代码实现

3.1 暴力解法

    int InversePairs(vector<int>& nums) {
        // write code here
        long long p = 0;
        int n = nums.size();
        for(int i = 0; i+1 < n; i++)
        {
            for(int j = i+1; j < n; j++)
            {
                if(nums[i] > nums[j])
                {
                    p++;
                }
            }
        }
        return fmod(p, 1000000007);
    }

3.2 归并排序法

class Solution {
private:
    const int kmod = 1000000007;
public:
    int InversePairs(vector<int> data) {
        int ret = 0;
        merge_sort__(data, 0, data.size() - 1, ret);
        return ret;
    }


    void merge_sort__(vector<int> &arr, int l, int r, int &ret) {
        if (l >= r) {
            return;
        }

        int mid = l + ((r - l) >> 1);
        merge_sort__(arr, l, mid, ret);
        merge_sort__(arr, mid + 1, r, ret);
        merge__(arr, l, mid, r, ret);
    }

    void merge__(vector<int> &arr, int l, int mid, int r, int &ret) {
        vector<int> tmp(r - l + 1);
        int i = l, j = mid + 1, k = 0;

        while (i <= mid && j <= r) {
            if (arr[i] > arr[j]) {
                tmp[k++] = arr[j++];
                // 奥妙之处
                ret += (mid - i + 1);
                ret %= kmod;
            }
            else {
                tmp[k++] = arr[i++];
            }
        }

        while (i <= mid) {
            tmp[k++] = arr[i++];
        }
        while (j <= r) {
            tmp[k++] = arr[j++];
        }

        for (k = 0, i = l; i <= r; ++i, ++k) {
            arr[i] = tmp[k];
        }
    }
};

4 代码解析

4.1 暴力解法

4.1.1 初始化
        long long p = 0;
        int n = nums.size();

初始化,p用来计算逆序对个数,n为循环边界。 

4.1.2 判断是否是逆序对
        for(int i = 0; i+1 < n; i++)
        {
            for(int j = i+1; j < n; j++)
            {
                if(nums[i] > nums[j])
                {
                    p++;
                }
            }
        }

使用两层嵌套循环统计逆序对,外层循环 i遍历数组的前 n-1 个元素。内层循环 ji+1 开始,遍历 i 之后的元素,检查 nums[i] > nums[j] 是否成立。如果满足 nums[i] > nums[j],则 p++ 记录一个逆序对。

4.2 归并排序法

4.2.1 InversePairs 主函数
int InversePairs(vector<int> data) {
    int ret = 0;
    merge_sort__(data, 0, data.size() - 1, ret);
    return ret;
}

ret 变量用于存储逆序对的个数。调用 merge_sort__data 进行归并排序,同时统计逆序对。

4.2.2 归并排序 merge_sort__ 递归拆分
void merge_sort__(vector<int> &arr, int l, int r, int &ret) {
    if (l >= r) {
        return;
    }

    int mid = l + ((r - l) >> 1);
    merge_sort__(arr, l, mid, ret);
    merge_sort__(arr, mid + 1, r, ret);
    merge__(arr, l, mid, r, ret);
}

该函数不断递归拆分数组,直到 l >= r(即子数组长度为 1)。递归调用 merge_sort__ 对左右子数组分别排序。最后调用 merge__ 进行合并并计算逆序对。 

4.2.3 归并 merge__ 统计逆序对
void merge__(vector<int> &arr, int l, int mid, int r, int &ret) {
    vector<int> tmp(r - l + 1);
    int i = l, j = mid + 1, k = 0;

    while (i <= mid && j <= r) {
        if (arr[i] > arr[j]) {
            tmp[k++] = arr[j++];
            // 统计逆序对数量
            ret += (mid - i + 1);
            ret %= kmod;
        } else {
            tmp[k++] = arr[i++];
        }
    }

    while (i <= mid) tmp[k++] = arr[i++];
    while (j <= r) tmp[k++] = arr[j++];

    for (k = 0, i = l; i <= r; ++i, ++k) {
        arr[i] = tmp[k];
    }
}

该方法利用双指针合并两个有序子数组,并在合并过程中统计逆序对。指针 i 遍历左半部分(lmid),j 遍历右半部分(mid + 1r)。当 arr[i] > arr[j] 时,说明 arr[i] 及其后续所有元素(mid - i + 1 个)都大于 arr[j],构成逆序对,累加 ret 并对 1000000007 取模防止溢出。排序时,较小的元素先存入 tmp,保持整体有序,最终将 tmp 复制回 arr,完成归并排序。

5 总结

本文介绍了两种计算数组逆序对的方法:暴力解法和归并排序法。暴力解法使用两层嵌套循环,逐一比较元素,时间复杂度为O(n^2),适用于小规模数据。归并排序法利用归并排序的分治思想,结合双指针合并两个有序子数组,并在合并过程中统计逆序对,时间复杂度降为O(n log n),适用于大规模数据。具体实现中,递归地拆分数组,合并时通过比较左右子数组元素,累加逆序对数量。归并排序法高效、稳定,且满足题目对时间和空间的复杂度要求,是解决该问题的推荐方案。


http://www.kler.cn/a/573447.html

相关文章:

  • BasicToolNode(tools=[search_tool, lookup_policy, query_sqldb])的内部执行逻辑
  • Android ChatOn-v1.66.536-598-[构建于ChatGPT和GPT-4o之上]
  • 大模型巅峰对决:DeepSeek vs GPT-4/Claude/PaLM-2 全面对比与核心差异揭秘
  • FastExcel/EasyExcel简介以及源码解析
  • 张岳教授:语言模型推理与泛化研究 | ICLR 2025 特邀报告与团队专场
  • 18类创新平台培育入库!长沙经开区2025年各类科技创新平台培育申报流程时间材料及申报条件
  • jQuery UI 简介
  • 【漫话机器学习系列】119.小批量随机梯度方法
  • 机器学习中的优化方法:从局部探索到全局约束
  • Measuring short-form factuality in large language models (SimpleQA) 论文简介
  • mybatis日期格式与字符串不匹配bug
  • 解锁前端表单数据的秘密旅程:从后端到用户选择!✨
  • 微服务通信:用gRPC + Protobuf 构建高效API
  • Java+SpringBoot+Vue+数据可视化的百草园化妆服务平台(程序+论文+讲解+安装+调试+售后)
  • 年后寒假总结及计划安排
  • linux安装Kafka以及windows安装Kafka和常见问题解决
  • 迷你世界脚本对象库接口:ObjectLib
  • Oracle CBD结构和Non-CBD结构区别
  • 微软官宣5 月 5 日关闭 Skype,赢者通吃法则依然有效
  • 解锁网络防御新思维:D3FEND 五大策略如何对抗 ATTCK