题目描述
给你两个有序整数数组 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]
算法1
(双指针) $O(n)$
从nums数组后面开始遍历,比较大小填入就行了
C++ 代码
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i = m - 1;
int j = n - 1;
int len = m + n - 1;
while(i >= 0 && j >= 0){
nums1[len --] = nums1[i] <= nums2[j] ? nums2[j--]:nums1[i--];
}
while(i >= 0) nums1[len --] = nums1[i --]; // 可能有剩余的
while(j >= 0) nums1[len --] = nums2[j --]; // 可能有剩余的
}
};