给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
解法:
1、使用C++自带的sort函数,先将nums2的值都复制到nums1的后面,然后直接使用sort函数进行排序即可
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
//将两个数组合并后再排序使用sort函数
vector<int> vec;
for(int i = 0;i < m;i++) vec.push_back(nums1[i]);
nums1.clear();
nums1 = vec;
for(int i = 0; i < n;i++)nums1.push_back(nums2[i]);
sort(nums1.begin(),nums1.end());
}
};
2、双指针法
(1)从前到后
使用两个指针p1,p2分别指向数组nums1和nums2,然后另开辟一个数组vec,指针p指向vec,将nums1的值复制过去,然后比较nums1[p1]和nums2[p2]大小并赋值给vec[p]
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
//双指针法,从前往后
int p1 = 0,p2 = 0,p = 0;
vector<int> vec(nums1);
while( (p1 < m)&&(p2 < n) )
{
nums1[p++] = vec[p1] <= nums2[p2] ? vec[p1++]:nums2[p2++];
}
if(p2 < n)
{
while(p2 < n)
{
nums1[p++] = nums2[p2++];
}
}
if(p1 < m)
{
while(p1 < m)
{
nums1[p++] = vec[p1++];
}
}
}
};
(2)从后到前
扫描二维码关注公众号,回复:
9143655 查看本文章
从nums1和nums2数组从后向前开始扫描,因为nums1数组后面没有值,所以这种方法可以节省空间。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
//双指针法,从后往前
int p1 = m-1;//nums1的指针
int p2 = n-1;//nums2的指针
int p = m + n -1;
while( (p1>=0)&&(p2>=0) )
{
nums1[p--] = nums1[p1] >= nums2[p2] ? nums1[p1--]:nums2[p2--];
}
if(p1 < 0)
{ //当nums1数组排完而nums2元素未被合并完
while(p2 >= 0)
{
nums1[p--] = nums2[p2--];
}
}
}
};