88. 合并两个有序数组
程序员文章站
2024-02-25 09:20:04
...
给定两个有序整数数组 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]
思路:
经典归并算法,懒得动脑,没省空间,下一个思路是个抖机灵方法。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
{
vector<int>temp;
vector<int>n1(nums1.begin(), nums1.begin() + m);
vector<int>n2(nums2.begin(), nums2.begin() + n);
int p1 = 0, p2 = 0;
while (p1<n1.size()&&p2<n2.size())
{
if (n1[p1] > n2[p2])
temp.push_back(n2[p2++]);
else
temp.push_back(n1[p1++]);
}
while (p1 < n1.size())
temp.push_back(n1[p1++]);
while (p2 < n2.size())
temp.push_back(n2[p2++]);
nums1 = temp;
}
};
思路2:
鉴于本题的描述,我们可以把nums2塞到nums1后面,然后再把nums1排序即可,(正经脸)
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
{
for (int i = 0; i < n; ++i)
{
nums1.at(m + i) = nums2.at(i);
}
sort(nums1.begin(), nums1.begin() + m + n);
}
};
上一篇: leetcode
下一篇: 674. 最长连续递增序列