力扣88题:合并两个有序数组

力扣88题:合并两个有序数组

题目描述

给定两个按非递减顺序排列的整数数组 nums1nums2,以及它们的长度 mn,要求将 nums2 合并到 nums1,使得合并后的数组仍按非递减顺序排列。

输入与输出

示例 1

plaintext 复制代码
输入:nums1 = [1,2,3,0,0,0], m = 3
      nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]

示例 2

plaintext 复制代码
输入:nums1 = [1], m = 1
      nums2 = [], n = 0
输出:[1]

示例 3

plaintext 复制代码
输入:nums1 = [0], m = 0
      nums2 = [1], n = 1
输出:[1]

算法思路

1. 问题分析

题目要求我们原地合并两个数组:

  • nums1 的后半部分预留了足够的空间(大小为 m + n m + n m+n)。
  • nums1nums2 已经是有序的。

2. 双指针逆向合并

我们从两个数组的尾部开始比较,选择较大的元素放入 nums1 的末尾。具体步骤如下:

2.1 初始化指针
  • 定义指针 p1:指向 nums1 的有效元素的末尾(即索引 m − 1 m - 1 m−1)。
  • 定义指针 p2:指向 nums2 的末尾(即索引 n − 1 n - 1 n−1)。
  • 定义指针 p:指向 nums1 的总末尾(即索引 m + n − 1 m + n - 1 m+n−1)。
2.2 比较与插入
  • 如果 nums1[p1] > nums2[p2],将 nums1[p1] 放入 nums1[p],并移动 p1p
  • 如果 nums1[p1] <= nums2[p2],将 nums2[p2] 放入 nums1[p],并移动 p2p
2.3 拷贝剩余元素
  • 如果 nums2 中还有未处理的元素,直接将它们拷贝到 nums1 的前面。
  • 如果 nums1 中还有未处理的元素,则无需额外操作。
2.4 循环终止条件
  • p1 < 0p2 < 0 时,循环结束。

代码实现

以下是修正后的完整代码:

c 复制代码
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n) {
    int i = nums1Size - 1; // 从 nums1 的尾部开始填充
    while (m > 0 || n > 0) {
        if (n > 0 && (m == 0 || nums1[m - 1] < nums2[n - 1])) {
            nums1[i--] = nums2[--n];
        } else {
            nums1[i--] = nums1[--m];
        }
    }
}

代码详解

1. 初始化指针

定义三个指针:

  • p1 = m - 1:指向 nums1 有效部分的末尾。
  • p2 = n - 1:指向 nums2 的末尾。
  • p = m + n - 1:指向 nums1 的尾部。

2. 从尾部向前合并

通过比较 nums1[p1]nums2[p2],将较大的元素放入 nums1[p],并更新指针。以下是操作逻辑:

c 复制代码
if (n > 0 && (m == 0 || nums1[m - 1] < nums2[n - 1])) {
    nums1[p--] = nums2[--n];
} else {
    nums1[p--] = nums1[--m];
}

3. 拷贝剩余的 nums2

如果 nums2 中还有未处理的元素,直接拷贝:

c 复制代码
while (n > 0) {
    nums1[p--] = nums2[--n];
}

复杂度分析

时间复杂度

  • 遍历数组时,每次比较、移动只需 O ( 1 ) O(1) O(1) 时间,总体复杂度为 O ( m + n ) O(m + n) O(m+n)。

空间复杂度

  • 使用了常量级的额外空间,复杂度为 O ( 1 ) O(1) O(1)。

测试用例

测试用例 1

输入

plaintext 复制代码
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3

输出

plaintext 复制代码
[1,2,2,3,5,6]

测试用例 2

输入

plaintext 复制代码
nums1 = [1], m = 1
nums2 = [], n = 0

输出

plaintext 复制代码
[1]

测试用例 3

输入

plaintext 复制代码
nums1 = [0], m = 0
nums2 = [1], n = 1

输出

plaintext 复制代码
[1]

测试用例 4

输入

plaintext 复制代码
nums1 = [2,2,2,0,0,0], m = 3
nums2 = [2,2,2], n = 3

输出

plaintext 复制代码
[2,2,2,2,2,2]
相关推荐
LWing6512 分钟前
算法基础学习|02归并排序——分治
c++·学习·算法
Normal Developer2 小时前
HotSpot的算法细节
java·jvm·算法
wuqingshun3141594 小时前
蓝桥杯 3. 压缩字符串
数据结构·c++·算法·职场和发展·蓝桥杯
Tanecious.8 小时前
初阶数据结构--排序算法(全解析!!!)
数据结构·算法·排序算法
hu_yuchen10 小时前
C++:Lambda表达式
开发语言·c++·算法
烨然若神人~10 小时前
算法训练营第五天 | 454.四数相加II\ 383. 赎金信\15. 三数之和\ 18. 四数之和
算法
一只鱼^_10 小时前
牛客周赛 Round 91
数据结构·c++·算法·数学建模·面试·贪心算法·动态规划
啊阿狸不会拉杆10 小时前
人工智能数学基础(三):微积分初步
人工智能·python·算法·数学建模
2401_8582861110 小时前
CC52.【C++ Cont】滑动窗口
开发语言·数据结构·c++·算法·leetcode·滑动窗口
mvufi11 小时前
day31 第八章 贪心算法 part05
算法·贪心算法