题目链接:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
解题思路:
因为num1只有前m个元素是有效元素,num1和num2有序,所以可以使用双指针进行比较,另外还可以利用num1后半部分可以使用的这部分空间,实现时间复杂度为O(m+n),空间复杂度为O(1)的算法,具体步骤如下:
- 如果 m = 0,此时,直接将num2中的元素复制到num1中即可,然后直接return
- 否则:每次从num1的有效部分的最后和num2的最后选择最大的元素,放在num1后面,这样可以防止num1中前面的元素被覆盖,
- 令 i = m-1,j= n-1,cnt = m+n-1
- 如果 i >0并且 j>=0,循环选择最大的元素放在后面:
- 如果num1[i] >= num2[j]:num1[cnt--] = num1[i--]
- 否则 num1[cnt--] = num2[j--]
- 如果 j >=0,说明num2中还有元素,将num2中剩余的元素放在num1中,循环执行:num1[cnt--] = num2[j--]
AC代码:
class Solution {
public static void merge(int[] nums1, int m, int[] nums2, int n) {
if (m == 0) {
System.arraycopy(nums2, 0, nums1, 0, n);
return;
}
int i = m - 1;
int j = n - 1;
int cnt = m + n - 1;
while (i >= 0 && j >= 0) {
if (nums1[i] >= nums2[j]) {
nums1[cnt--] = nums1[i--];
} else {
nums1[cnt--] = nums2[j--];
}
}
while (j>=0){
nums1[cnt--]=nums2[j--];
}
}
}