【leetcode】88. 合并两个有序数组( Merge Sorted Array )

---

题目描述

【leetcode】88. 合并两个有序数组( Merge Sorted Array )
给定两个有序整数数组 nums1 和 nums2，将 nums2 合并到 nums1 中，使得 num1 成为一个有序数组。

说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间（空间大小大于或等于 m + n）来保存 nums2 中的元素。

第一次解答

思路：
nums1已经有足够的空间，
数组从小到大排序
为了防止大量的内存拷贝，从后往前添加元素

test case:
[1,2,3,0,0,0]
3
[2,5,6]
3
[4,5,6,0,0,0]
3
[1,2,3]
3

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        
        // 序号指向两个数组末尾
        int index_1 = m-1;
        int index_2 = n-1;
        for(int i=m+n-1; i>=0; --i){
            // 如果一个数组序号小于0，那只添加另一个数组元素
            if(0 > index_1){
                nums1[i] = nums2[index_2--];
                continue;
            }
            if(0 > index_2){
                nums1[i] = nums1[index_1--];
                continue;
            }

            // 若序号都大于0，挑选数值大的元素添加
            if(nums1[index_1] > nums2[index_2]){
                nums1[i] = nums1[index_1--];
            }
            else{
                nums1[i] = nums2[index_2--];
            }

        }
    }
};

结果：

第二次解答

思路：
方法与第一次解答一样，不变，优化代码减少if判断次数。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        
        // 序号指向两个数组末尾
        int index_1 = m-1;
        int index_2 = n-1;
        int index = m+n-1;

        while(0 <= index_1 && 0 <= index_2){
            // 若序号都大于0，挑选数值大的元素添加
            nums1[index--] = (nums1[index_1] > nums2[index_2] ? nums1[index_1--] : nums2[index_2--]);
        }
        // 到这里后，若0 > index_2，则merge完成
        // 若0 > index_1，则把nums2剩下元素放到nums1中即可
        while(0 <= index_2){
            nums1[index--] = nums2[index_2--];
        }
    }
};

结果：

相关/参考链接