88.合并两个有序数组

Posted 2020-12-10 Jessey45

题目描述： 给你两个有序整数数组 nums1 和 nums2，请你将 nums2 合并到 nums1 中，使 num1 成为一个有序数组。
说明:
初始化 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]

先合并再排序：

//C 手写快排

void partition(int* arr, int low, int high);
int quickSort(int* arr, int low, int high);

void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
    //先合并 再快排
    int i;
    for(i = 0; i < n; i++) nums1[m + i] = nums2[i];
    partition(nums1, 0, m + n - 1);
}
int quickSort(int* arr, int low, int high){
    int tmp = arr[low];
    while(low < high){
        while(low < high && arr[high] >= tmp) high--;
        arr[low] = arr[high];
        while(low < high && arr[low] <= tmp) low++;
        arr[high] = arr[low];
    }
    arr[low] = tmp;
    return low;
}

void partition(int* arr, int low, int high){
    if(low < high){
        int mid = quickSort(arr, low, high);
        partition(arr, low, mid - 1);
        partition(arr, mid + 1, high);
    }
}

或者使用语言自带的排序函数也可。

//C

int comp(const void *a, const void *b){
    return *(int*)a - *(int*)b;
}

void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
    int i;
    for(i = 0; i < n; i++) nums1[m + i] = nums2[i];
    qsort(nums1, m + n, sizeof(int), comp);
}

//JS

var merge = function(nums1, m, nums2, n) {
    nums1.splice(m,n,...nums2);
    return nums1.sort((a, b) => a - b);
};

双指针，从前往后遍历：使用额外的数组空间，时间复杂度O(n + m), 空间复杂度O(m)

//C

void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
        int *tmp_arr = (int *)malloc(sizeof(int) * m);
        int i, p1, p2;
        p1 = p2 =0;
        //
        for(i = 0; i < m; i++) tmp_arr[i] = nums1[i];
        i = 0;
        while(p1 < m && p2 < n) nums1[i++] = tmp_arr[p1] < nums2[p2] ? tmp_arr[p1++] : nums2[p2++];
        while(p1 < m) nums1[i++] = tmp_arr[p1++];
        while(p2 < n) nums1[i++] = nums2[p2++];
}

双指针，从后往前遍历：不需要使用额外的数组空间，时间复杂度O(n + m), 空间复杂度O(1)

//C

void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
        int i = m + n - 1,
        p1 = m - 1,
        p2 = n - 1;
        while(p1 >= 0 && p2 >= 0) nums1[i--] = nums1[p1] > nums2[p2] ? nums1[p1--] : nums2[p2--];
        while(p1 >= 0) nums1[i--] = nums1[p1--];
        while(p2 >= 0) nums1[i--] = nums2[p2--];
}