给你两个按 非递减顺序 排列的整数数组
nums1
和nums2
,另有两个整数m
和n
,分别表示nums1
和nums2
中的元素数目。请你 合并
nums2
到nums1
中,使合并后的数组同样按 非递减顺序 排列。注意:最终,合并后数组不应由函数返回,而是存储在数组
nums1
中。为了应对这种情况,nums1
的初始长度为m + n
,其中前m
个元素表示应合并的元素,后n
个元素为0
,应忽略。nums2
的长度为n
。示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3 输出:[1,2,2,3,5,6] 解释:需要合并 [1,2,3] 和 [2,5,6] 。 合并结果是 [1,2,2,3,5,6] ,其中斜体加粗标注的为 nums1 中的元素。
思路:简单的双指针,从小到大遍历
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i=0,j=0;
vector<int> t;
while(i<m||j<n){
if(i==m){
t.push_back(nums2[j]);
j++;
}
else if(j==n){
t.push_back(nums1[i]);
i++;
}
else if(nums1[i]<=nums2[j]){
t.push_back(nums1[i]);
i++;
}
else{
t.push_back(nums2[j]);
j++;
}
}
for(int i=0;i<m+n;i++)
nums1[i]=t[i];
}
};
记录下高畅大佬的做法,直接从大到小,减少了空间
我们直接利用 m 和 n 当作两个数组的指针,再额外创立一个 pos 指针,起始位置为 m + n − 1。每次向前移动 m 或 n 的时候,也要向前移动 pos。
这里需要注意,如果 nums1的数字已经复制完,不要忘记把 nums2 的数字继续复制;如果 nums2 的数字已经复制完,剩余nums1 的数字不需要改变,因为它们已经被排好序。
注意 这里我们使用了 ++ 和--的小技巧:a++ 和 ++a 都是将 a 加 1,但是 a++ 返回值为 a,而
++a 返回值为 a+1。如果只是希望增加 a 的值,而不需要返回值,则推荐使用 ++a,其运行速度
会略快一些。
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i=m-1,j=n-1;
int pos=m+n-1;
while(i>=0&&j>=0){
if(nums1[i]>=nums2[j]){
nums1[pos--]=nums1[i--];
}
else{
nums1[pos--]=nums2[j--];
}
}
while(j>=0){
nums1[pos--]=nums2[j--];
}
}
//大佬代码
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int pos = m-- + n-- - 1;
while (m >= 0 && n >= 0) {
nums1[pos--] = nums1[m] > nums2[n]? nums1[m--]: nums2[n--];
}
while (n >= 0) {
nums1[pos--] = nums2[n--];
}
}