leetcode---C++实现---88. Merge Sorted Array（合并两个有序数组）

题目

level：easy
Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:

• The number of elements initialized in nums1 and nums2 are m and n respectively.
• You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.

Example:

Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
Output: [1,2,2,3,5,6]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/merge-sorted-array

解题思路

由于需要将两个数组中的元素合并到一个数组中，两个数组均已排序完毕，为避免出现往数组中插入元素导致数组后面元素挪动的情况，比较时两数组从后往前遍历。

1. 声明三个变量，int i 记录数组nums1有效数字的下标（初始化为 m-1），int j 记录数组nums2的下标（初始化为 n-1）， int index记录返回数组nums1的下标（初始化为 m+n-1）；
2. 比较 nums1[i] 与 nums2[j] 的大小，则将两者中较大的数赋值给 nums1[index]，再将较大数的数组记录的下标减1， index 也减1，继续比较nums1中元素与nums2中元素大小；
3. 当某一个数组全部遍历完毕，则不再进行比较。若剩余nums1的元素，则不用做任何事，因为nums1中的元素已在nums1且已排序；若剩余nums2的元素，则需将nums2中剩余的元素继续赋值到nums1中index所指的位置。

算法实现（C++）

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) { 
        int index = m + n - 1;
        int i = m -1, j = n -1;
        while (i >= 0 && j >= 0)
        {
            if (nums1[i] > nums2[j])
            {
                nums1[index--] = nums1[i--];
            }
            else
            {
                nums1[index--] = nums2[j--];
            }
        }

        while (j >= 0)
            nums1[index--] = nums2[j--];
    }
};

复杂度分析

• 时间复杂度：O(m+n)，每次遍历即给返回数组的一个元素赋值，一共有 m+n 个元素；
• 空间复杂度：O(1)，定义了三个变量记录下标，常数空间。