问题描述 链接到标题

88.合并两个有序数组

解题思路 链接到标题

双指针,由于num1的长度是m + n,因此遍历时,可以考虑针对nums1,从后往前赋值,从而只需要$\Theta(1)$的额外空间。

代码 链接到标题

#include <vector>
using std::vector;
class Solution {
    public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int p1 = m - 1, p2 = n - 1;
        int idx = m + n - 1;
        for (int idx = m + n - 1; idx >= 0; idx--) {
            if (p1 < 0)
                nums1[idx] =  nums2[p2--];
            else if (p1 >= 0 && p2 >= 0) {
                if (nums1[p1] > nums2[p2])
                    nums1[idx] = nums1[p1--];
                else
                    nums1[idx] = nums2[p2--];
            }
        }
    }
};