题目描述
Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.
Note:
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2. The number of elements initialized in nums1 and nums2 are m and n respectively.
思路
给两个排好序的整型数组nums1和nums2,将nums2的值归并到nums1,并使之仍然有序。
设一个新的数组存放结果,最后再全部赋给nums1。
设两个下标i,j,分别表示两个数组的当前位置,比较后,哪个小就赋给新的数组,那么哪个就向后移动。注意的是一个结束了,另一个数组里还有剩余的值也要全部存入新的数组里。
代码
class Solution {
public:
vector<int> merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
vector<int> nums3;
int i, j;
for(i = 0, j = 0; i < m && j < n;)
{
if(nums1[i] < nums2[j])
{
nums3.push_back(nums1[i]);
i++;
}
else
{
nums3.push_back(nums2[j]);
j++;
}
}
while(i < m)
{
nums3.push_back(nums1[i]);
i++;
}
while(j < n)
{
nums3.push_back(nums2[j]);
j++;
}
nums1.clear();
for(i = 0; i < nums3.size(); i++)
{
nums1.push_back(nums3[i]);
}
return nums1;
}
};