从前面移动元素的话会非常麻烦,所以这道题目的思路是从数组的后面开始移动元素,将两个数组从大到小移动而不是从前面从小到大移动,这种方法非常巧妙,也是一种常用的方法,每当从前面移动数组或字符串不是很方便的时候可以思考可否从数组的后面进行移动。另外当某个数组的元素全都移动完毕之后,我们需要将另一个数组剩下的元素进行移动,这里利用了题目的一个特性,就是我们移动元素的目标是nums1,也就是说,如果nums1的长度比较大的话,那么我们在退出循环之后什么都不用做,因为前面的元素不用移动也已经in-place sorted,所以只用检测nums2是否移动完即可。时间复杂度O(n + m),代码如下;
public class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
// Set pointers to end of array
int i = m - 1;
int j = n - 1;
int k = m + n - 1;
// Move elements from end of array
while(i >= 0 && j >= 0){
// Move larger element to end of nums1
if(nums1[i] > nums2[j]){
nums1[k] = nums1[i];
i--;
k--;
}else{
nums1[k] = nums2[j];
j--;
k--;
}
}
// Move remaining elements to nums1, if m > n, no need to move
while(j >= 0){
nums1[k] = nums2[j];
j--;
k--;
}
}
}
知识点:
1. 记住这种从后面移动数组的思想,每当从前方操作数组或是字符串比较麻烦的时候,可以尝试下从后方移动是否可行