题解:给定两个排序的整型数组nums1和nums2,将nums2合并到nums1成一个排序数组。你可以假设nums1中有足够的空间(空间大于或等于m+n)来存放来自nums2的额外元素。nums1和nums2的初始空间分别是m和n。
分析:题目本身比较简单,要求in-place处理数组,因此不能另外申请一个m+n的数组nums3。好在nums1有足够的空间,但是如果从数组头部开始merge的话,可能会覆盖nums1中原来的元素。
因此,知道最终合并的长度为m+n以后,从nums1的尾部开始merge,读取nums1[m]和nums[n]的数值,比较谁大就把谁添加到nums1的m+n位,然后在逐步向前推进。
时间复杂度O(m+n),空间复杂度为常数O(1)。
import java.util.Arrays;
class Solution {
public static void merge(int[] nums1, int m, int[] nums2, int n) {
int i = m - 1, j = n - 1;
for(; i >= 0 && j >= 0;){
if(nums1[i] > nums2[j]){
nums1[i + j + 1] = nums1[i];
i--;
}else{
nums1[i + j + 1] = nums2[j];
j--;
}
}
// 需要注意,当上面循环结束时,nums2有可能还有元素没有merge到nums1中,因此需要单独处理一下。
while(j >= 0){
nums1[j] = nums2[j];
j--;
}
}
public static void main(String[] args) {
int[] nums1 = {1,2,3,0,0,0,0};
int[] nums2 = {4,5,6,7};
merge(nums1, 3, nums2, 4);
System.out.println(Arrays.toString(nums1));
}
}