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

浙公网安备 33010602011771号