算法-合并两个有序数组

20 篇文章 0 订阅
5 篇文章 0 订阅

1. 合并两个有序数组

给定两个有序整数数组 nums1nums2,将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组

初始化 nums1nums2 的元素数量分别为 m 和 n ,此处假设 nums1空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素

LeetCode 链接

示例 :

输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]

2. 解法

比较容易想到的解法有如下两个:

  1. 直接合并两个数组,然后对数组进行排序,返回排序后的数组即可
  2. 从前往后遍历两个数组,依次比较两数组对应元素值大小,每次比较后将最小值放入额外数组中,最后返回这个额外数组即可

以上第 2 种解法在时间复杂度上已经是最优的,但是在空间复杂度上还有优化空间。另一种解法是从后往前遍历两个数组,使用 3 个指针来记录数组关键位置下标,依次比较更新即可

    public void merge(int[] nums1, int m, int[] nums2, int n) {
        // 使用 3 个指针记录数组关键下标:
        // p1 -> nums1 数组尾部
        // p2 -> nums2 数组尾部
        // p -> 最终数组尾部
        // 1.当 p1 >= 0 时,nums1[p1], nums2[p2]对比
        //     A. nums1[p1]大,将 nums1[p1] 放入 nums1[p] 位置, p1--, p--
        //     B. nums2[p2]大于等于 nums1[p1],将nums2[p2]放入 nums1[p] 位置, p2--, p--
        // 2.当 p1 < 0 时,将 nums2[p2]放入 nums1[p] 位置, p2--,p--
        // 循环结束:p2<0

        int p1 = m - 1;
        int p2 = n - 1;
        int p = m + n - 1;
        while (p2 >= 0) {
            if (p1 >= 0 && nums1[p1] > nums2[p2]) {
                nums1[p--] = nums1[p1--];
            } else {
                nums1[p--] = nums2[p2--];
            }
        }
    }
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值