&LeetCode88& 合并两个有序数组

题目

给定两个有序整数数组 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--];
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章