合并两个有序数组
题目:
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
说明:
初始化 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]
方法一:指针法
先分别将nums1和nums2排序,指针p1指向nums1的首部,遍历nums1,指针p2指向nums2,若*p1<=p2,p1++,否则将p2插入nums1,p2++。
代码一:
class Solution
{
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
{
sort(nums1.begin(),nums1.begin()+m);
sort(nums2.begin(), nums2.begin()+n);
nums1.erase(nums1.begin() + m, nums1.end());
nums2.erase(nums2.begin() + n, nums2.end());
int i = 0, j = 0;
for (;i<m,j<n;)
{
if (nums1[i] <= nums2[j])
{
nums1.push_back(nums1[i]);
i++;
}
else
{
nums1.push_back(nums2[j]);
j++;
}
}
if (i == m)
{
while (j<n)
{
nums1.push_back(nums2[j]);
j++;
}
}
else if (j == n)
{
while (i<m)
{
nums1.push_back(nums1[i]);
i++;
}
}
nums1.erase(nums1.begin(), nums1.begin()+m+1);
}
};//结果可能超时
方法二:合并后排序
先讲两个数组合并,再使用sort函数排序,简单且直观。
注意:合并nums1和nums2并且排序之后,需要删掉nums1尾部的0。
代码二:
class Solution
{
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
{
nums1.insert(nums1.begin()+m,nums2.begin(),nums2.begin()+n);
sort(nums1.begin(),nums1.begin()+m+n);
nums1.erase(nums1.begin()+m+n,nums1.end());
}
};