合并两个有序数组

LeetCode第88题

题目描述：

给你两个有序整数数组 nums1 和 nums2，请你将 nums2 合并到 nums1 中，使 nums1 成为一个有序数组。

说明:

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。
你可以假设 nums1 有足够的空间（空间大小大于或等于 m + n）来保存 nums2 中的元素。

示例:

输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3

输出: [1,2,2,3,5,6]

思路

一看到题目立马想到从前往后遍历，将较小的值保存在nums1中。但是这样有问题，可能会将nums1中还未处理的值覆盖。
所以就想到额外创建一个大小为m+n的数组result，来存储结果，最后将result数组拷贝给nums1。
后来看题解，发现还可以从后往前遍历，将较大的值保存在nums1中（按下标从大往小存储）。这样就不需要额外的数组空间了。因为nums1中，最初的空余空间就是n个，从后往前存储时，不会覆盖nums1中未处理的数值。

源代码

//开辟额外空间
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
    int k = 0;
    int i = 0;
    int j = 0;
    int * result = (int *)malloc(sizeof(int) * (m + n));

    while(i < m && j < n){
        if(nums1[i] <= nums2[j]){
            result[k++] = nums1[i++];
        }
        else{
            result[k++] = nums2[j++];
        }
    }

    while(i < m){
        result[k++] = nums1[i++];
    }

    while(j < n){
        result[k++] = nums2[j++];
    }

    for(k = 0;k < (m + n);k++)
        nums1[k] = result[k];
}

//不开辟额外空间
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
    int k = m + n - 1;
    int i = m - 1;
    int j = n - 1;

    while(i >= 0 && j >= 0){
        if(nums1[i] >= nums2[j]){
            nums1[k--] = nums1[i--];
        }
        else{
            nums1[k--] = nums2[j--];
        }
    }

    while(j >= 0){
        nums1[k--] = nums2[j--];
    }
}

分析

时间复杂度为O（m+n）；
空间复杂度解法一为O（m+n），解法二为O（1）。