力扣(LeetCode)88. 合并两个有序数组(C++)

news2025/2/26 22:54:43

朴素思想

朴素思想，开第三个数组，对 $n u m s 1$ 和 $n u m s 2$ 进行二路归并。

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

时间复杂度: $O (n + m)$ ， $n$ 是 $n u m s 1$ 的长度， $m$ 是 $n u m s 2$ 的长度，遍历所有数的时间复杂度是 $O (n + m)$ 。

空间复杂度: $O (n + m)$ ， $n u m s 3$ 的空间复杂度是 $O (n + m)$ 。

逆序归并

由于 $n u m s 1 . s i z e () = m + n$ ，思考倒着归并。设极端情况， $n u m s 2$ 的元素全部大于 $n u m s 1$ ，需要把 $n u m s 2$ 接到 $n u m s 1$ 后面， $n u m s 1$ 前面 $m$ 个数是 $n u m s 1$ 本身的数，后面 $n$ 个数是 $n u m s 2$ 的数， $m + n = n u m s 1 . s i z e ()$ ，这种极端情况也可以容纳。那么正常归并时，如果选择 $n u m s 1$ 的数，那么 $n u m s 1$ 最后的数就可以被覆盖了，这种情况同样可以容纳。

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