1.合并两个有序数组

题目

给你两个按非递减顺序排列的整数数组nums1和nums2，另有两个整数m和n，分别表示nums1和nums2中的元素数目。

请你合并nums2到nums1中，使合并后的数组同样按非递减顺序排列。

注意：最终，合并后数组不应由函数返回，而是存储在数组nums1中。为了应对这种情况，nums1的初始长度为m + n，其中前m个元素表示应合并的元素，后n个元素为0，应忽略。nums2的长度为n。

示例 1：

输入：nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3

输出：[1,2,2,3,5,6]

解释：需要合并 [1,2,3] 和 [2,5,6] 。

合并结果是 [1,2,2,3,5,6] ，其中斜体加粗标注的为 nums1 中的元素。

示例 2:

输入：nums1 = [1], m = 1, nums2 = [], n = 0

输出：[1]

解释：需要合并 [1] 和 [] 。

合并结果是 [1] 。

示例 3：

输入：nums1 = [0], m = 0, nums2 = [1], n = 1

输出：[1]

解释：需要合并的数组是 [] 和 [1] 。

合并结果是 [1] 。

注意，因为 m = 0 ，所以 nums1 中没有元素。nums1 中仅存的 0 仅仅是为了确保合并结果可以顺利存放到 nums1 中。

思路

逆向双指针:
- nums1中是非递减的数组。 - nums2中也是非递减数组。 - 所以我们要做的就是把nums1中前m个元素与nums2中的元素进行倒序比较，将大的值倒序放到nums1中的后面。 - 因为nums1中的数据已经是非递减的了，所以等nums2中的内容都放置完就可以结束。

class Solution:
    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
        index = m + n - 1
        p1 = m - 1
        p2 = n - 1
        while p2 >= 0:
            if p1 >= 0 and nums1[p1] >= nums2[p2]:
                nums1[index] = nums1[p1]
                index += 1
                p1 -= 1
            else:
                nums1[index] = nums2[p2]
                index += 1
                p2 -= 1

class Solution {
    fun merge(nums1: IntArray, m: Int, nums2: IntArray, n: Int): Unit {
        var index = nums1.lastIndex
        var r1 = m - 1
        var r2 = n - 1
        while(r2 >= 0) {
            if (r1 >= 0 && nums1[r1] >= nums2[r2]) {
                nums1[index --] = nums1[r1--]
            } else {
                nums1[index --] = nums2[r2 --]
            } 
        }
    }
}

- 下一篇

• 邮箱 ：charon.chui@gmail.com
• Good Luck!