题目描述: 给你两个有序整数数组 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--]; }