1. 合并两个有序数组
给定两个有序整数数组 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]
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
2. 解法
比较容易想到的解法有如下两个:
- 直接合并两个数组,然后对数组进行排序,返回排序后的数组即可
- 从前往后遍历两个数组,依次比较两数组对应元素值大小,每次比较后将最小值放入额外数组中,最后返回这个额外数组即可
以上第 2 种解法在时间复杂度上已经是最优的,但是在空间复杂度上还有优化空间。另一种解法是从后往前遍历两个数组,使用 3 个指针来记录数组关键位置下标,依次比较更新即可
public void merge(int[] nums1, int m, int[] nums2, int n) {
// 使用 3 个指针记录数组关键下标:
// p1 -> nums1 数组尾部
// p2 -> nums2 数组尾部
// p -> 最终数组尾部
// 1.当 p1 >= 0 时,nums1[p1], nums2[p2]对比
// A. nums1[p1]大,将 nums1[p1] 放入 nums1[p] 位置, p1--, p--
// B. nums2[p2]大于等于 nums1[p1],将nums2[p2]放入 nums1[p] 位置, p2--, p--
// 2.当 p1 < 0 时,将 nums2[p2]放入 nums1[p] 位置, p2--,p--
// 循环结束:p2<0
int p1 = m - 1;
int p2 = n - 1;
int p = m + n - 1;
while (p2 >= 0) {
if (p1 >= 0 && nums1[p1] > nums2[p2]) {
nums1[p--] = nums1[p1--];
} else {
nums1[p--] = nums2[p2--];
}
}
}