题目
给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。
示例
题解
解法一 双指针,创建一个新数组,while循环,每次把较小的数存入新数组中,最后复制给num1数组
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i = 0;
int j = 0;
vector<int> ans;
while(i < m && j < n){
if(nums1[i] <= nums2[j]){
ans.push_back(nums1[i]);
i++;
}
else{
ans.push_back(nums2[j]);
j++;
}
}
if(i < m){
for(int p = i;p < m; p++)
ans.push_back(nums1[p]);
}
if(j < n){
for(int p = j;p < n; p++)
ans.push_back(nums2[p]);
}
nums1.assign(ans.begin(),ans.end());
}
};
解法二
因为这两个数组已经排好序,我们可以把两个指针分别放在两个数组的末尾,即 nums1 的m − 1 位和 nums2 的 n − 1 位。每次将较大的那个数字复制到 nums1 的后边,然后向前移动一位。
因为我们也要定位 nums1 的末尾,所以我们还需要第三个指针,以便复制。
在以下的代码里,我们直接利用 m 和 n 当作两个数组的指针,再额外创立一个 pos 指针,起始位置为 m + n − 1。每次向前移动 m 或 n 的时候,也要向前移动 pos。这里需要注意,如果 nums1的数字已经复制完,不要忘记把 nums2 的数字继续复制;如果 nums2 的数字已经复制完,剩余nums1 的数字不需要改变,因为它们已经被排好序。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i = m - 1;
int j = n - 1;
int pos = m + n -1;
while(i >= 0 && j >= 0){
if(nums1[i] > nums2[j]){
nums1[pos] = nums1[i];
pos--;i--;
}
else{
nums1[pos] = nums2[j];
pos--;j--;
}
}
while(j >= 0){
nums1[pos] = nums2[j];
pos--;j--;
}
}
};