合并两个有序数组

 

目录

题目：

解法1：双指针法

解法2：暴力法

解法3：逆向双指针

---

 

题目：

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

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n，这样它就有足够的空间保存来自 nums2 的元素。

 

示例 1：

输入：nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出：[1,2,2,3,5,6]
示例 2：

输入：nums1 = [1], m = 1, nums2 = [], n = 0
输出：[1]
 

 

解法1：双指针法

两个数组中从第一个元素之间对比，将最小元素加入到temp数组中，然后依次往后对比，当某一个数组已经走完之后，将另一个数组其余元素追加到temp中。

注：vector中没有“/0”。

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

特征：两有序数组+排序                                                   解法：双指针解法

 

解法2：暴力法

将nums2数组内容放到nums1中，然后直接排序。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        for (int i = 0; i != n; ++i) {
            nums1[m + i] = nums2[i];
        }
        sort(nums1.begin(), nums1.end());
    }
};

特征：无序数组                                                                解法：sort()

 

解法3：逆向双指针

思路：因为nums1数组长度为m+n大小，所以可以将两数组内容从后向前对比，这样可以防止数组在赋值时内容被覆盖。

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

            nums1[i--] = cur;
        }
        
    }
};

优化：

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

 

特征：一个数组可以存所两个数组的所有内容+两数组都是递增数组                 解法：后序向前赋值