力扣 题目88- 合并两个有序数组

题目

题解

这种题 一看 我们直接先从后面向前想

很容易能想到我们谁大取谁 这里的向后 分别是指m n 以及遍历的nums1->p= m+ n - 1;

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

一开始 

nums1 [m]=3 nums2[n] =6 所以nums1 [p]=6;->p--;n-- ->1 2 3 0 0 6

nums1 [m]=3 nums2[n] =5 所以nums1 [p]=5;->p--;n-- ->1 2 3 0 5 6

nums1 [m]=3 nums2[n] =2 所以nums1 [p]=3;->p--;m-- ->1 2 3 3 5 6

nums1 [m]=2 nums2[n] =2 所以nums1 [p]=2;->p--;n-- ->1 2 2 3 5 6

即可

        int p = m-- + n-- - 1;
        while (m >= 0 && n >= 0) {
            nums1[p--] = nums1[m] > nums2[n] ? nums1[m--] : nums2[n--];
        }

但是我们发现 如果nums2 的值比nums1 的第一个值小呢?那么最后nums2 剩下的都是小的

那么我们直接nums1[p--] = nums2[n--];

        //这里是比nums1的第一个还小的 nums2数值直接替换前面的数字即可
        while (n >= 0) {
            nums1[p--] = nums2[n--];
        }

当然 其实这题从前往后也可以做 但是很多条件没用到 说明这题的本意不是这个 所以这里不说了 有兴趣可以看代码

代码

后->前

#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int p = m-- + n-- - 1;
        while (m >= 0 && n >= 0) {
            nums1[p--] = nums1[m] > nums2[n] ? nums1[m--] : nums2[n--];
        }

        while (n >= 0) {
            nums1[p--] = nums2[n--];
        }
       }
};

int main() {
    Solution sol;
    vector<int> nums1 = { 2,0 }; 
    int m = 1; 
    vector<int> nums2 = { 1 };
    int n = 1;
    sol.merge(nums1,m, nums2,n);
    for (int i = 0; i < nums1.size(); i++) {
        cout << nums1[i] << " ";
    }
}

前->后

#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        if (n == 0) {
            return;
        }
        int ergodic = 0;
        for (int i = 0; i < nums1.size()&& ergodic<n; i++) {
            if (nums1[i] == 0&&i>m-1) {
                nums1[i] = nums2[ergodic];
                ergodic++;   
            }
            else if(nums1[i] >= nums2[ergodic])
            {
                nums1.insert(nums1.begin()+i, nums2[ergodic]);
                nums1.pop_back();
                ergodic++;
                m++;
            }
        }
    }
};

int main() {
    Solution sol;
    vector<int> nums1 = { 2,0 }; 
    int m = 1; 
    vector<int> nums2 = { 1 };
    int n = 1;
    sol.merge(nums1,m, nums2,n);
    for (int i = 0; i < nums1.size(); i++) {
        cout << nums1[i] << " ";
    }
}