[leetcode] 88. Merge Sorted Array

Description

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

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

Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
Output: [1,2,2,3,5,6]

Analysis

果然还是自己太菜，想的方法特别绕，一看discuss里面大佬的解法才恍然大悟。
大概思路就是，从后向前进行插入，先定一个总的下标k=m+n-1，也就是两个数组合到一起之后的最后一个元素下标。nums1和nums2也都从最后一个元素开始，分别为i=m-1，j=n-1。每次比较nums1和nums2中当前的元素，将大的那个放在k下标处，并将对应下标-1。
一个细节是，i和j都必须>=0。而如果遇到这种情况：当i=0且j=0且nums1[i]=nums2[j]时，将nums2[j]插入k下标位置，然后j–。注意此时j=-1，不符合j>=0的条件了，因此循环会结束。但nums1[0]本身就在剩下的那个位置上了，就不用再插入了。

Code

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(j>=0){
            if(i<0){
                nums1[k--] = nums2[j--];
            }
            else if(j<0){
                break;
            }
            if(i>=0 && nums1[i]>nums2[j]){
                nums1[k--] = nums1[i--];
            }
            else{
                nums1[k--] = nums2[j--];
            }
        }
    }
};