LeetCode（88）Merge Sorted Array

题目

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:

You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2. The number of elements initialized in nums1 and nums2 are m and n respectively.

分析

题目输入为两个vector有序集合，要求合并两个集合（不可删除其中元素，就是说重复元素都保存在结果中），结果保存在第一个参数集合中。

本题程序实现，借用了另一个vector的空间用于保存合并后的集合，最后将此集合赋值给第一个参数集合即可。

考虑：最优算法应该是本地合并而不需要任何其他空间。

AC代码

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

        while (i < m)
        {
            ret.push_back(nums1[i]);
            i++;
        }//while

        while (j < n)
        {
            ret.push_back(nums2[j]);
            j++;
        }//while

        nums1 = ret;

    }

};

GitHub测试程序源码