leetcode_双指针 88.合并两个有序数组

原创于 2025-02-02 23:18:29 发布 · 215 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

88.合并两个有序数组

给你两个按非递减顺序排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。
请你合并 nums2 到 nums1 中，使合并后的数组同样按非递减顺序排列。
注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。
思路：
- 由于 nums1 末尾已经有足够的空间（初始长度为 m+n），可以从后向前进行合并，避免额外的移动操作。故方法为比较 nums1 和 nums2 的末尾元素，并把较大的元素放到 nums1 的最后位置

class Solution(object):
    def merge(self, nums1, m, nums2, n):
        """
        :type nums1: List[int]
        :type m: int
        :type nums2: List[int]
        :type n: int
        :rtype: None Do not return anything, modify nums1 in-place instead.
        """
        # 初始化三个指针i,j,k, i指向nums1的最后一个有效元素, j指向nums2的最后一个有效元素, k指向nums1的末尾
        i, j, k = m - 1, n - 1, m + n - 1

        while i >= 0 and j >= 0:
            if nums1[i] > nums2[j]:
                nums1[k] = nums1[i]
                i -= 1
            else:
                nums1[k] = nums2[j]
                j -= 1
            k -= 1

        # 若 nums2 还有剩余元素，则继续填充
        while j >= 0:
            nums1[k] = nums2[j]
            j -= 1
            k -= 1