力扣88——合并两个有序数组

题目描述
给定两个有序数组,把两个数组合并为一个。

输入输出样例
输入是两个数组和它们分别的长度 m 和 n。其中第一个数组的长度被延长至 m + n,多出的
n 位被 0 填补。题目要求把第二个数组归并到第一个数组上,不需要开辟额外空间。

Input: nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
Output: nums1 = [1,2,2,3,5,6]

题解
因为这两个数组已经排好序,我们可以把两个指针分别放在两个数组的末尾,即 nums1 的 m − 1 位和 nums2 的 n − 1 位。每次将较大的那个数字复制到 nums1 的后边,然后向前移动一位。

因为我们也要定位 nums1 的末尾,所以我们还需要第三个指针,以便复制。

在以下的代码里,我们直接利用 m 和 n 当作两个数组的指针,再额外创立一个 pos 指针,起始位置为 m +n−1。每次向前移动 m 或 n 的时候,也要向前移动 pos。
这里需要注意,如果 nums1的数字已经复制完,不要忘记把 nums2 的数字继续复制;
如果 nums2 的数字已经复制完,剩余nums1 的数字不需要改变,因为它们已经被排好序。

这里我们使用了 ++ 和–的小技巧:a++ 和 ++a 都是将 a 加 1,但是 a++ 返回值为 a,而++a 返回值为 a+1。如果只是希望增加 a 的值,而不需要返回值,则推荐使用 ++a,其运行速度会略快一些

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int pos=m-- + n-- - 1;
        while(m >= 0 && n >=0){
            nums1[pos--]=nums1[m]>nums2[n]?nums1[m--]:nums2[n--];
        } 
        while(n>=0){
            nums1[pos--]=nums2[n--];
        }
    }
};



力扣88——合并两个有序数组_第1张图片

你可能感兴趣的