乍一看像是归并排序,但是从题目来看明显是可以使用原地算法的。
使用额外数组的方法很简单:
void merge(int* nums1, int m, int* nums2, int n) {
int * temp = malloc(sizeof(int)*(m+n));
int i=0,j = 0;
while(i<m&&j<n){
temp[i+j] = nums1[i]<nums2[j]?nums1[i++]:nums2[j++];
}
while(i<m){
temp[i+j] = nums1[i++];
}
while(j<n){
temp[i+j] = nums2[j++];
}
for(i=0;i<m+n;i++){
nums1[i] = temp[i];
}
}
简单的两路归并。
但是需要注意到题目的条件,nums1的容量是大于等于m+n的,也就是说可以直接把结果放进m+n。
但如何在不影响nums1原本元素的情况下进行排序?nums1的前m个空间存着待排序元素,但后n个空间一定是空的。
那么如果从尾部开始填入排序的结果就不会有问题了,因为就算把nums2的n个元素全部塞进nums1尾部,
也不会挤到nums1自己的元素。
采用这样的方法相当于反向归并,从队尾开始,取较大的元素。
void merge(int* nums1, int m, int* nums2, int n) {
int idx = m + n - 1;
int i = m - 1;
int j = n - 1;
while(i >= 0 && j >= 0) {
int a = nums1[i], b = nums2[j];
if(a > b) {
i--;
nums1[idx--] = a;
} else {
j--;
nums1[idx--] = b;
}
}
while(i >= 0) {
nums1[idx--] = nums1[i--];
}
while(j >= 0) {
nums1[idx--] = nums2[j--];
}
}