Day45 合并两个有序数组

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

https://leetcode-cn.com/problems/merge-sorted-array/

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

示例1:

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

示例2:

输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]

提示:

nums1.length == m + n
nums2.length == n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[i] <= 109

Java解法

思路:

  • 一道简单题,不需要额外空间,直接在nums1上插入
  • 分别记录两数组遍历位置,插入进行排序
  • 一道简单题做起来也出错两次0.0
    • 注意输入参数都是必要参数,我以为全是正数,所以以0判断是不是到末尾了
package sj.shimmer.algorithm.m2;

import sj.shimmer.algorithm.Utils;

/**
 * Created by SJ on 2021/3/11.
 */

class D45 {
    public static void main(String[] args) {
//        int[] nums1 = {1, 2, 3, 0, 0, 0};
//        int[] nums2 = {2, 5, 6};
//        int[] nums1 = {2,0};
//        int[] nums2 = {1};

        int[] nums1 = {-1, 0, 0, 3, 3, 3, 0, 0, 0};
        int[] nums2 = {1, 2, 2};
        merge(nums1, 6, nums2, 3);
        Utils.logArray(nums1);
    }
    public static void merge(int[] nums1, int m, int[] nums2, int n) {
        int index1 = 0;
        int index2 = 0;
        int length = nums1.length;
        while (index1<length&&index2<n) {
            if (nums1[index1]<nums2[index2]) {
                if (index1>=m) {
                    nums1[index1]=nums2[index2];
                    m++;
                    index2++;
                }
                index1++;
            }else {
                //nums1后移
                for (int i = length -1; i >index1; i--) {
                    nums1[i]=nums1[i-1];
                }
                nums1[index1] =nums2[index2];
                m++;
                index2++;
                index1++;
            }
        }
    }
}

官方解

https://leetcode-cn.com/problems/merge-sorted-array/solution/he-bing-liang-ge-you-xu-shu-zu-by-leetcode/

  1. 合并后排序

    最简单直接的做法,时间复杂度差,但代码简单就两行

    System.arraycopy(nums2, 0, nums1, m, n);
    Arrays.sort(nums1);
    
    • 时间复杂度:O((n+m)log(n+m))
    • 空间复杂度: O(1)
  2. 双指针

    从前往后:用额外的M空间存储nums1的数据,双指针比较存入数组中,得到最优时间复杂度

    从后往前:不需要额外空间,只需要比较从末尾开始存储即可,NICE

    public void merge(int[] nums1, int m, int[] nums2, int n) {
        // two get pointers for nums1 and nums2
        int p1 = m - 1;
        int p2 = n - 1;
        // set pointer for nums1
        int p = m + n - 1;
        // while there are still elements to compare
        while ((p1 >= 0) && (p2 >= 0))
            // compare two elements from nums1 and nums2 
            // and add the largest one in nums1 
            nums1[p--] = (nums1[p1] < nums2[p2]) ? nums2[p2--] : nums1[p1--];
    
        // add missing elements from nums2
        System.arraycopy(nums2, 0, nums1, 0, p2 + 1);
    }
    
    • 时间复杂度:O(n+m)
    • 空间复杂度: O(1)
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值