给定两个有序整数数组 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]
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/merge-sorted-array 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
很简单的题目
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int n1 = nums1.size();
int idx1 = 0, idx2 = 0;
while(idx1 < m && idx2 < n)
{
if(nums1[idx1] < nums2[idx2])
nums1.push_back(nums1[idx1++]);
else
nums1.push_back(nums2[idx2++]);
}
if(idx1 == m)
{
while(idx2 < n)
nums1.push_back(nums2[idx2++]);
}
else//(idx2 == n)
{
while(idx1 < m)
nums1.push_back(nums1[idx1++]);
}
nums1.erase(nums1.begin(),nums1.begin()+n1);
}
};