给你两个有序整数数组 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++;
}
}
}
}
image
官方解
-
合并后排序
最简单直接的做法，时间复杂度差，但代码简单就两行
System.arraycopy(nums2, 0, nums1, m, n); Arrays.sort(nums1);- 时间复杂度：O((n+m)log(n+m))
- 空间复杂度： O(1)
-
双指针
从前往后：用额外的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)
