088 - Merge Sorted Array

解法

merge sort 的 “merge” step 實作。

程式碼

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int index_m, index_n;
        index_m = index_n = 0;

        vector<int> ans;
        while(1){
           //檢查 nums2 是否還有值 || nums1 <= nums2
           if(index_n >= n || (index_m < m && nums1[index_m] <= nums2[index_n])){
               ans.push_back(nums1[index_m]);
               index_m++;
           } //vice versa
           else if(index_m >= m || (index_n < n && nums2[index_n] < nums1[index_m])){
               ans.push_back(nums2[index_n]);
               index_n++;
           }

           if(index_n >= n && index_m >= m) break;
        }

        nums1.clear();
        for(auto it: ans){
            nums1.push_back(it);
        }
    }

};