合并两个有序数组
题目解析
题目要求对两个以按非递减排序的数组进行合并,合并成一个非递减排序数组。所以我们可以用双指针,分别指向两个数组最左端,依次比较,小的(或相等的)存入新建的一个结果数组,比较过元素后指针则后移。最后按照题目要求将结果数组存入nums1数组即可。
代码实现
初始化双指针i,j为0。当i<m且j<n时,即nums1与nums2数组均没有将所有元素都比较完时,两数组还有元素可以进行比较(如第一个循环所示),将比较小的先存入ret数组,对应的指针后移。当i或j有一个大于对应数组长度,说明已经有一个数组中的元素都已经比较完且合并入ret数组了,所以仅需对另一个数组中剩余元素接着依次存入ret数组即可。
最后,将ret数组赋值给nums1数组。
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
vector<int> ret;
int i = 0, j = 0;
while(i < m && j < n){
if(nums1[i]<nums2[j]){
ret.push_back(nums1[i]);
i++;
}else{
ret.push_back(nums2[j]);
j++;
}
}
while(i < m){
ret.push_back(nums1[i]);
i++;
}
while(j < n){
ret.push_back(nums2[j]);
j++;
}
nums1 = ret;
return;
}