Merge Sorted Array_LeetCode

Description:

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.

 

算法思想：

对两个数组同时从后向前遍历，从m+n-1的下标从后往前写入，比较两个数组同时所指数字的大小：

将大数写入，且下标-1；小数不写入，下标不动。

当出现nums1已经写完了，而nums2还有未写完的情况，继续写nums2进去。

 

代码：

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