题目描述:
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 and nums2 are m and n respectively.
解题思路:
这是一道数组操作的题目,思路比较明确,就是维护三个数组,分别对应数组A,数组B,和结果数组(长度为m+n)。然后A和B同时从后往前扫(从m+n开始往回),每次迭代中A和B指向的元素大的便加入结果数组中,然后index-1,另一个不动。
这里从后往前扫是因为这个题目中结果仍然放在A中,如果从前扫会有覆盖掉未检查的元素的可能性。算法的时间复杂度是O(m+n),m和n分别是两个数组的长度,空间复杂度是O(1)。
代码如下:
public void merge(int[] nums1, int m, int[] nums2, int n) {
int k = m + n - 1;
int i = m - 1;
int j = n - 1;
while (k >= 0) {
if (i < 0) {
nums1[k--] = nums2[j--];
continue;
}
if (j < 0) {
break;
}
if (nums1[i] > nums2[j]) {
nums1[k--] = nums1[i--];
} else {
nums1[k--] = nums2[j--];
}
}
}
下面这个方法在题设的情况下应该也对的,但是不知道为啥不让我通过。。。
修改之后发现是自己把nums1和nums2写错了所以数组越界了。。。
public void merge(int[] nums1, int m, int[] nums2, int n) {
if(nums1==null||nums2==null)
return ;
int idx1 = m-1;
int idx2 = n-1;
int len = m+n-1;
while(idx1>=0 && idx2>=0)
{
if(nums1[idx1]>nums2[idx2])
{
nums1[len--] = nums1[idx1--];
}
else
{
nums1[len--] = nums2[idx2--];
}
}
while(idx2>=0)
{
nums1[len--] = nums2[idx2--];
}
}