题目解析来自图解面试算法。
法一:合并后排序
最朴素的方法,将nums2数组复制到nums1里面,然后数组排序。
注意先是源数组,再是目标数组。
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
System.arraycopy(nums2, 0, nums1, m, n);
Arrays.sort(nums1);
}
}
时间复杂度:O( (m+n) log(m+n) )
空间复杂度:O(1)
法二:双指针 + 从前向后比较(即把小的放到指针p处)
public void merge (int[] nums1, int m, int[] nums2, int n){
// 复制nums1数组到一个新的位置
int[] nums1_copy = new int[m];
System.arraycopy(nums1, 0, nums1_copy, 0, m);
// 双指针
int p1 = 0;
int p2 = 0;
// nums1的指针,指向排序后数字的位置
int p = 0;
// 比较nums1_copy和nums2
// 将小的加到nums1中
while (p1<m && p2<n) {
nums1[p++] = (nums1_copy[p1] < nums2[p2]) ? nums1_copy[p1++] : nums2[p2++];
}
// 最后的检查
if (p1 < m) {
System.arraycopy(nums1_copy, p1, nums1, p1 + p2, m + n - p1 - p2);
}
if (p2 < n) {
System.arraycopy(nums2, p2, nums1, p1 + p2, m + n - p1 - p2);
}
}
时间复杂度:O (m+n)
空间复杂度:O(m)
法三:双指针 + 从后向前比较
注意指针的初始化是最后一个位置的索引,即长度-1。
public void merge (int[] nums1, int m, int[] nums2, int n){
// 两个数组的指针
int p1 = m-1;
int p2 = n-1;
// 指向nums1数组中最后一个数字,即存放排序后数字的位置
int p = m+n-1;
while (p1>=0 && p2>=0) {
nums1[p--] = (nums1[p1] < nums2[p2]) ? nums2[p2--] : nums1[p1--];
}
// 遍历后检查,若指向p2的指针还有效,说明nums2中还有小于nums1最小值的存在
// 需要将它们放到nums1最前面
System.arraycopy(nums2, 0, nums1, 0, p2 + 1);
}
时间复杂度:O(m+n)
空间复杂度:O(1)