力扣88. 合并两个有序数组----倒序遍历双指针

88. 合并两个有序数组

给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。

示例 1:

输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
示例 2:

输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
 

提示:

nums1.length == m + n
nums2.length == n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[i] <= 109

题解:

很明显我们直接采用双指针原理即可完成合并,即比对两个数组的一个数字,谁大取谁指针挪移即可。但是从头遍历的话会覆盖掉nums1的数据,因此我们需要再开辟一个数组。

方法一:额外数组+双指针(从前遍历)

代码:

void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
    
    
    int i = 0;
    int j = 0;
    int*res = (int*)malloc(sizeof(int)*(m+n));
    int k = 0;
    while(i<m||j<n)
    {
    
    
        if(i>=m)
        {
    
    
            res[k++] = nums2[j++];
            continue;
        }
        if(j>=n)
        {
    
    
            res[k++] = nums1[i++];
            continue;
        }
        if(nums1[i]<=nums2[j])
        {
    
    
            res[k++] = nums1[i++];
            continue;
        }
        else
        {
    
    
            res[k++] = nums2[j++];
        }
    }
    for(k=0;k<m+n;k++)
    {
    
    
        nums1[k] = res[k];
    }
    free(res);
}

方法二:原数组+双指针

我们可以直接从后遍历,这样就可以避免开辟新数组。

代码:

void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
    
    
    int i = m-1;
    int j = n-1;
    int k = m+n-1;
    while(i>=0||j>=0)
    {
    
    
        if(i<0)
        {
    
    
            nums1[k--] = nums2[j--];
            continue;
        }
        if(j<0)
        {
    
    
            nums1[k--] = nums1[i--];
            continue;
        }
        if(nums1[i]>=nums2[j])
        {
    
    
            nums1[k--] = nums1[i--];
            continue; 
        }
        if(nums1[i]<nums2[j])
        {
    
    
            nums1[k--] = nums2[j--];
            continue;
        }
    }
}

猜你喜欢

转载自blog.csdn.net/xiangguang_fight/article/details/115445067