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.

思路

给两个排好序的整型数组nums1和nums2，将nums2的值归并到nums1，并使之仍然有序。

设一个新的数组存放结果，最后再全部赋给nums1。

设两个下标i，j，分别表示两个数组的当前位置，比较后，哪个小就赋给新的数组，那么哪个就向后移动。注意的是一个结束了，另一个数组里还有剩余的值也要全部存入新的数组里。

代码

class Solution {
public:
    vector<int> merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        vector<int> nums3;
        int i, j;
        for(i = 0, j = 0; i < m && j < n;)
        {
            if(nums1[i] < nums2[j])
            {
                nums3.push_back(nums1[i]);
                i++;
            }
            else
            {
                nums3.push_back(nums2[j]);
                j++;
            }
        }
        while(i < m)
        {
            nums3.push_back(nums1[i]);
            i++;
        }
        while(j < n)
        {
            nums3.push_back(nums2[j]);
            j++;
        }
        nums1.clear();
        for(i = 0; i < nums3.size(); i++)
        {   
            nums1.push_back(nums3[i]);
        }
        return nums1;
    }
};