LeetCode面试经典150题—01 合并有序数组
思路
- 思路一：双指针 + 新建数组
这是很容易想到的一种思路，用双指针思想分别遍历比较两个数组中数的大小，将较小的那个数依次放入新创建的数组中，最后将该数组拷贝到 nums1
代码如下：
public static void merge(int[] nums1, int m, int[] nums2, int n) {
int[] ans = new int[m + n];
int p1 = 0, p2 = 0, p3 = 0;
while (p1 < m && p2 < n) {
if (nums1[p1] <= nums2[p2]) {
ans[p3] = nums1[p1];
p1++;
} else {
ans[p3] = nums2[p2];
p2++;
}
p3++;
}
if (p1 == m) {
for (int i = p2; i < n; i++) {
ans[p3] = nums2[i];
p3++;
}
} else {
for (int i = p1; i < m; i++) {
ans[p3] = nums1[i];
p3++;
}
}
System.arraycopy(ans, 0, nums1, 0, m + n);
}
问题：没有利用好 nums1 数组初始长度为 m + n 的特点，且新创建数组占用内存
优化：从后往前确认数组，直接在 nums1 上进行变化，注意 nums2 未完全插入的情况
代码如下：
public static void merge(int[] nums1, int m, int[] nums2, int n) {
m--;
n--;
while (m >= 0 && n >= 0) {
nums1[m + n + 1] = nums2[n] >= nums1[m] ? nums2[n--] : nums1[m--];
}
while (n >= 0) {
nums1[m + n + 1] = nums2[n--];
}
}
- 思路二：先合并再排序
这种思路简单且代码量比较少，但是没有利用到题目中数组 非递减顺序 的条件
代码如下：
public static void merge(int[] nums1, int m, int[] nums2, int n) {
System.arraycopy(nums2, 0, nums1, m, n);
Arrays.sort(nums1);
}
数组拷贝方法拓展
- System.arraycopy()
源码：
参数：
src：源数组
srcPos：源数组要复制的起始位置
dest：目标数组
destPos：目标数组要复制的起始位置
length：要复制的数组长度
用法：
public static void main(String[] args) {
int[] arr1 = new int[]{0, 1, 2, 5};
int[] arr2 = new int[4];
System.arraycopy(arr1, 0, arr2, 0, arr1.length);
System.out.println(Arrays.toString(arr2));
}
//输出：[0, 1, 2, 5]- Arrays.copyOf()
源码：
从源码可以看出，Arrays.copyOf() 实际上就是调用的 System.arraycopy()
用法：
public static void main(String[] args) {
int[] arr1 = new int[]{0, 1, 2, 5};
int[] arr2 = Arrays.copyOf(arr1, arr1.length);
System.out.println(Arrays.toString(arr2));
}
//输出：[0, 1, 2, 5]- clone()
用法：
public static void main(String[] args) {
int[] arr1 = new int[]{0, 1, 2, 5};
int[] arr2 = arr1.clone();
System.out.println(Arrays.toString(arr2));
}
//输出：[0, 1, 2, 5]
注意在思路一中是不能用 clone() 来将新创建的数组拷贝给 nums1 的，这里的拷贝是给形参 nums1 重新赋值，改变了引用对象，并不会改变实参 nums1
这点与 Java 的值传递有关，可参考视频进行理解：
Java 到底是值传递还是引用传递
关于 clone() 的更多内容可参考文章：
Java 中 clone 方法详解