题目描述:
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 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]
这道题其实自己写有很多实现方法,用一个新的数组可以很简单的实现这个目标,但是我们既然去为它写个笔记,那自然不会写一个十分常规的办法:
至简代码:
function merge(nums1, m, nums2, n){
var p = m-- + n-- -1;
while(m>=0 && n>=0){
nums1[p--] = nums1[m]>nums2[n]?nums1[m--]:nums2[n--];
}
while(n>=0){
nums1[p--] = nums2[n--];
}
return nums1;
}
可以看到,这种方法并没有使用新数组。但时间和空间效率也可以达到94%、97%以上。
这个方法最重要的思路就是从后向前更新主数组,有什么优点呢?
用这个方法有一个前提,那就是主数组的长度得可以包含两个数组实例元素。空的位可以用0临时占位。
从后向前遍历主数组的好处:可以避免从前向后遍历时,辅数组元素全部替代了主数组元素,因而主数组元素无法被找到的情况。
从后向前就可以保证:即使是主数组的元素被移动,也始被移动到主数组中,而不会丢失。
其次就是对于第二个while循环,为什么只需要考虑n就够了?
仍然以题干中的这个例子为例,当nums2中的n遍历到-1时,就说明nums2已经被遍历完了,这时nums1中的m还指向nums1[1],接下来,代码执行到第二个while循环,发现n并不≥0,于是直接跳过while循环,然后return。
不需要分析m吗?并不需要,为什么?当n执行到-1时,说明了nums2中的所有元素都比当前m所指向的num1中的元素大,在nums1中,其实nums2中的元素在此时都已经被拼接到了m后面正确的位置。而m及之前的元素,自然都是比nums2小的元素,直接保留就可以了。