LeetCode

88. 合并两个有序数组

2018-12-02  本文已影响7人  闭门造折

给定两个有序整数数组 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]

思路

从后往前遍历两个数组,选择较大的放到nums1尾部空间中

性能分析

时间复杂度O(N),空间复杂度O(1)

具体代码

void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
    // 使用题目给的m和n作为尾索引
    m--;
    n--;
    // 使用r(ight)表示新数组尾索引
    int r = m + n + 1;
    // 当两个数组均有剩余时
    while(m >=0 && n >= 0){
        // 选取两个尾部较大的一个,放入新数组
        if(nums1[m] > nums2[n]){
            nums1[r--] = nums1[m--];
        }
        else{
            nums1[r--] = nums2[n--];
        }
    }
    // 若nums1先全部插入完毕,则nums2剩余部分全部复制
    if(m == -1){
        while(n>= 0){
            nums1[r--] = nums2[n--];
        }
    }
    else{
        while(m >= 0){
            nums1[r--] = nums1[m--];
        }
    }
    return;
}
};
上一篇下一篇

猜你喜欢

热点阅读