这道题的核心思想:
利用数组已经有序的特点,采用"从后往前填充"的贪心思想,每次选择当前最大的元素放到合并数组的最后位置。
因为 nums1 后面有预留空间,如果从前往后合并,会导致原来的元素被覆盖,需要不断移动数据。因此选择逆向遍历:
从 nums1 和 nums2 的最后一个有效元素开始比较;
当前两个元素中较大的那个,一定应该放在合并数组的最后空位;
放入后,继续向前比较剩余元素;
直到所有元素都移动到正确位置。
本质上就是:
利用有序数组的性质,每次确定当前最大值的位置,通过双指针从尾部向前构造最终有序数组。
这种方法避免了额外空间,也避免了元素移动,所以能够在 O(m+n) 时间复杂度、O(1) 空间复杂度 下完成合并。
cpp
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n) {
int i=m-1;
int j=n-1;
int k=m+n-1;
while(i>=0&&j>=0)
{
if(nums1[i]>nums2[j])
{
nums1[k--]=nums1[i--];
}
else
{
nums1[k--]=nums2[j--];
}
}
while(i>=0)
{
nums1[k--]=nums1[i--];
}
while(j>=0)
{
nums1[k--]=nums2[j--];
}
}