跳转至

88. 合并两个有序数组

难度:简单

题目

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

说明:

  • 初始化 nums1nums2 的元素数量分别为 mn
  • 你可以假设 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]

Reference

题解

基本思路: 直接在nums1的内存空间上从后向前原位合并。

代码如下:

#define MAX(x, y) (x > y ? x : y)
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
    int x = m - 1, y = n - 1;
    while(y >= 0 && x >= 0)
    {
        nums1[x + y + 1] = MAX(nums1[x], nums2[y]);
        if (nums1[x] > nums2[y])
            x--;
        else
            y--;
    }
    if (y >= 0)
        memcpy(nums1, nums2, sizeof(int) * (y + 1));
}

评论