Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.
Note:
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2. The number of elements initialized in nums1 andnums2 are m and n respectively.
----------------------------------------------------------------------------------------------------------------------------------------
我们需要明白,nums1中是有足够的空间,归并后的结果的。
在这里我讲介绍2种解法,一种是我最初想到过的空间复杂度为O ( m ) 的解法。另一种是比较好的inplace的解法。
两者的时间复杂度都是O(m + n )
第一种:
再开辟一个数组空间num1Copy,放nums1的前[0 ... m ] 的值,
然后对num1Copy 和 nums2进行归并,将结果放到nums1中去。
运行时间:
代码:
public void merge(int[] nums1, int m, int[] nums2, int n) {
int[] nums1Copy = new int[m];
for (int i = 0; i < m; i++) {
nums1Copy[i] = nums1[i];
}
int i = 0, j = 0;
int count = 0;
while (i < m && j < n) {
if (nums1Copy[i] < nums2[j]) {
nums1[count++] = nums1Copy[i++];
} else {
nums1[count++] = nums2[j++];
}
}
if (i < m) {
for (; i < m; i++) {
nums1[count++] = nums1Copy[i];
}
} else if (j < n) {
for (; j < n; j++) {
nums1[count++] = nums2[j];
}
}
}
另一种inplace的方法。
非常巧妙,从后往前,归并值大的的。(防止未排序的数组值被更新)
运行时间:
代码:
public void merge2(int[] nums1, int m, int[] nums2, int n) {
int i = m - 1, j = n - 1, k = m + n - 1;
while (i >= 0 && j >= 0) {
nums1[k--] = nums1[i] > nums2[j] ? nums1[i--] : nums2[j--];
}
while (j >= 0) {
nums1[k--] = nums2[j--];
}
}
参考资料:
https://leetcode.com/discuss/31675/share-my-accepted-java-solution