1、题目描述
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
2、示例
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
3、题解
基本思想:双指针,从后往前遍历,因为nums1后面多余空闲位置,这样就不需要申请额外空间。
i从后往前指向nums1有效元素,j从后往前指向nums2有效元素,cur从m+n-1到0指向合并的元素。
#include<iostream>
#include<vector>
#include<algorithm>
#include<iterator>
using namespace std;
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
//基本思想:双指针,从后往前遍历,因为nums1后面多余空闲位置,这样就不需要申请额外空间
//i从后往前指向nums1有效元素,j从后往前指向nums2有效元素,cur从m+n-1到0指向合并的元素
int i = m - 1, j = n - 1, cur = m + n - 1;
while (i >= 0 && j >= 0)
{
if (nums1[i] < nums2[j])
nums1[cur--] = nums2[j--];
else
nums1[cur--] = nums1[i--];
}
while (i >= 0)
nums1[cur--] = nums1[i--];
while (j >= 0)
nums1[cur--] = nums2[j--];
}
};
int main()
{
Solution solute;
vector<int> nums1 = { 2,0 };
vector<int> nums2 = { 1 };
int m = 1, n = 1;
solute.merge(nums1, m, nums2, n);
copy(nums1.begin(), nums1.end(), ostream_iterator<int>(cout, " "));
return 0;
}