LeetCode腾讯精选练习50题-88.合并两个有序数组

2022-05-04
阅读量 | 字数：623字 | 预计阅读时长：2分钟

题目描述

给定两个按非递减顺序排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。
请合并 nums2 到 nums1 中，使合并后的数组同样按非递减顺序排列。
注意：
最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。
为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。
nums2 的长度为 n 。

example
input  : nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
output : [1,2,2,3,5,6]
input  : nums1 = [1], m = 1, nums2 = [], n = 0
output : [1]
note   :

解题思路

思路1

遍历数组nums2，将其元素逐个放入nums1初始的后n个位置
对数组nums1排序
时间复杂度：O(N×log(N))，其中 N = m+n
空间复杂度：O(1)

思路2

双指针，同时逆序遍历数组nums1和nums2，其中nums1从第m个元素遍历到第1个元素，nums2从第n个元素遍历到第1个元素
假设当前nums1被遍历的元素为 a， nums2被遍历的元素为b，则取a和b中的较大者，放入nums1的最后
即从nums1的真正末尾位置 m+n 开始向前填充值，将遍历过程中m和n的较大者逐个放入nums1的后方，并且后方存放元素的指针不断前移
因为nums1和nums2本身是有序递增的，所以填充得到的结果依然是有序的
若nums1遍历结束，而nums2未结束，则继续遍历nums2并将其所有元素逐个移动到nums1的剩余位置中
当nums2遍历结束，而nums1未结束，则不需要继续操作，此时未遍历完的nums1的部分元素一定还是有序递增的排列在nums1的前方，没有被变动过，且一定都小于其当前的所有后方元素
时间复杂度：O(N)，其中 N = m+n
空间复杂度：O(1)

代码（Java）

思路1代码

public class Solution1 {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        for (int i = 0; i < n; i++) {
            nums1[i + m] = nums2[i];
        }
        Arrays.sort(nums1);
    }
}

思路2代码

public class Solution2 {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int len = m + n - 1;
        int i = m - 1, j = n - 1;
        while (i >=0 && j >=0){
            nums1[len--] = nums1[i] > nums2[j] ? nums1[i--] : nums2[j--];
        }
        while(j >= 0){
            nums1[len--] = nums2[j--];
        }
    }
}

本文作者： whtli
本文链接： https://hexo.whtli.cn/archives/daef219.html
版权声明： 遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。