[LeetCode][初级算法][排序] 合并两个有序数组

https://leetcode-cn.com/explore/interview/card/top-interview-questions-easy/8/sorting-and-searching/52/

乍一看像是归并排序,但是从题目来看明显是可以使用原地算法的。

使用额外数组的方法很简单:

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--];
    }
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值