题目
给定两个有序整数数组 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)
思路
本题中,可以从 nums1 和 nums2 数组的末尾开始逐个比较,将较大的数,从后往前加入混合数组的末尾;
首先,需要三个变量 i,j,k,分别指向 nums1,nums2,和混合数组的末尾;
其次,进行 while 循环,若i和j都大于0,再看如果 nums1[i] > nums2[j],说明要先把 nums1[i] 加入混合数组的末尾,加入后k和i都要自减1;
再次,若nums1[i] < nums2[j],就把 nums2[j] 加入混合数组的末尾,加入后k和j都要自减1;
循环结束后,可能i或者j还大于等于0,若j大于0,那么还需要继续循环,将 nums2 中的数字继续拷入 nums1中;
若是i大于等于0,不用管,因为混合数组本身就放在 nums1 中的。
C++代码
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
{
int i = m - 1, j = n - 1, k = m + 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--];
}
};