合并两个有序数组

1
2
3
4
5
6
给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。

请你 合并 nums2 到 nums1 中，使合并后的数组同样按 非递减顺序 排列。

注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。

时空复杂度均为O(n+m)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

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

优化空间复杂度

注意到nums1的初始长度为m+n，设置p指针指向nums1最后一个位置。

倒序遍历nums1和nums2数组，将二者较大的值倒序放入nums1。

倒序可以保证原数组值不被覆盖。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int p = m+n-1;
        --m;
        --n;
        while(m>-1&&n>-1){
            if(nums1[m]>nums2[n]) nums1[p--]=nums1[m--];
            else nums1[p--]=nums2[n--];
        }
        while(m>-1) nums1[p--]=nums1[m--];
        while(n>-1) nums1[p--]=nums2[n--];
    }
};