题目描述
思路
因为nums1保证空间足够,且nums1和nums2都是有序数组,可以考虑从后向前遍历两个数组,这样可以避免从前向后时nums1数组因为排序导致的交换问题。
代码实现1
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
// 2个数组一共m+n个元素,合并之后最大索引即为m+n-1
int max_index = m + n - 1;
int i = m -1, j = n - 1; // 从后向前遍历
while(i >= 0 || j >= 0) {
if (i >= 0 && j >= 0) {
nums1[max_index--] = nums1[i] >= nums2[j] ? nums1[i--] : nums2[j--];
} else if (i >= 0) {
nums1[max_index--] = nums1[i--];
} else {
nums1[max_index--] = nums2[j--];
}
}
}
};
复制代码
代码实现2
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int max_index = m + n - 1;
m--; n--;
// 这里不用或的条件,而用与,即先遍历结束其中一个数组
// 如果是nums1先结束,则继续nums2,也就是后一个while循环
// 如果是nums2先结束,nums1不用继续遍历,结果本身就是保存在nums1里,这就是最终结果了
while(m >= 0 && n >= 0) {
nums1[max_index--] = nums1[m] >= nums2[n] ? nums1[m--] : nums2[n--];
}
while(n >= 0) {
nums1[max_index--] = nums2[n--];
}
}
};
复制代码