leetcode 88. 合并两个有序数组

一、题目

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

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 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]

二、解法

解法1：使用临时数组+双指针

思路：
用一个临时数组保存nums1中的前m个数据，然后用双指针，从前往后遍历，遇到小的就放入nums1中，如果有一个数组提前遍历完，就把另一个数组的剩余部分全部放入nums1中。

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

    }
};

解法2：解法1的优化

解法1的空间复杂度是$O(m)$，其实是可以降低到$O(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;
        while(i>=0||j>=0){
            if(i<0){
                while(j>=0)nums1[k--]=nums2[j--];
                break;
            }
            if(j<0){
                while(i>=0)nums1[k--]=nums1[i--];
                break;
            }
            if(nums1[i]>nums2[j]){
                nums1[k--]=nums1[i--];
            }else{
                nums1[k--]=nums2[j--];
            }
        }

    }
};