问题描述 链接到标题
解题思路 链接到标题
双指针,由于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--];
}
}
}
};