【每日算法】LeetCode 88 —— 合并两个有序数组(一百八十)

题目内容

给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。

示例

示例 1:

输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
示例 2:

输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]

提示

1、nums1.length == m + n
2、nums2.length == n
3、0 <= m, n <= 200
4、1 <= m + n <= 200
5、-10^9 <= nums1[i], nums2[i] <= 10^9

题解

本题采用双指针算法。题目中强调nums1的空间较大,因此我们可以采用nums1来存储最终答案。具体做法如下:

1、我们从后往前遍历,这样能够保证,nums1中的元素不被覆盖掉

2、然后i和j两个指针指向两个数组的末尾,i指向nums1的第m-1的位置,j指向nums2的第n-1的位置

3、然后比较两个指针指代的数的大小,大的放置到nums1的最后,然后对应指针向前移动,持续比较,直到两个数组合并为有序数组为止

代码

class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
//将答案存入第一个数组中
int k = n + m - 1;//k来存储开始的下标
int i = m - 1, j = n - 1;
while(i >= 0 && j >= 0){
if(nums1[i] >= nums2[j])nums1[k--] = nums1[i--];
else nums1[k--] = nums2[j--];
}
while(j >= 0)nums1[k--] = nums2[j--];
}
};
Author: Frederic Niu
Link: https://www.fredericniu.cn/2021/06/25/【每日算法】LeetCode-88-——-合并两个有序数组(一百八十)/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.
我的公众号