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]
题解
因为这两个数组已经排好序,我们可以把两个指针分别放在两个数组的末尾,即 nums1 的 m - 1 位和 nums2 的 n - 1 位。每次将较大的那个数字复制到 nums1 的后边,然后向前移动一位。
因为我们也要定位 nums1 的末尾,所以我们还需要第三个指针,以便复制。
在以下的代码里,我们直接利用 m 和 n 当作两个数组的指针,再额外创立一个 pos 指针,起始位置为 m +nn1。每次向前移动 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) { // pos 新数组的位置 int pos = m-- + n-- - 1; /* 这里就是 int pos = m + n - 1; m--,n--; 的意思 */ while(m>=0&&n>=0) { nums1[pos--] = nums1[m] > nums2[n] ? nums1[m--]: nums2[n--]; /* 这里就是 nums1[pos] = nums1[m] > nums2[n] ? nums1[m]: nums2[n]; pos--,m--,n--; 的意思 */ } // 当 nums2 空了 nums1 不用去管 前面小的数据已经排好序了 ; // 当 nums1 空了 就直接把 nums2 加到数组中 while(n>=0) nums1[pos--]=nums2[n--]; } };