给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
法一:逻辑,先把nums2放到nums1后面,再排序(不推荐)
时间复杂度 : O((n+m)log(n+m))。
空间复杂度 : O(1)。
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
for (int i=m, j=0; i<m+n; j++,i++) {
nums1[i] = nums2[j];
}
Arrays.sort(nums1);
}
}
法二:从后往前插入(推荐)
因为 nums1 的空间都集中在后面,所以从后向前处理排序的数据会更好,节省空间,一边遍历一边将值填充进去
设置指针 len1 和 len2 分别指向 nums1 和 nums2 的有数字尾部,从尾部值开始比较遍历,同时设置指针 len 指向 nums1 的最末 尾,每次遍历比较值大小之后,则进行填充
当 len1<0 时遍历结束,此时 nums2 中海油数据未拷贝完全,将其直接拷贝到 nums1 的前面,最后得到结果数组
时间复杂度:O(m+n)
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int n1 = m-1;
int n2 = n-1;
int p = nums1.length-1;
while (n1 >=0 && n2 >=0) {
if (nums1[n1] > nums2[n2]) {
nums1[p] = nums1[n1];
p--;
n1--;
} else {
nums1[p] = nums2[n2];
p--;
n2--;
}
}
while (n2 >= 0) {
nums1[p] = nums2[n2];
p--;
n2--;
}
}
}