Python赋值 & 双指针法合并有序数组

写作目的:理解双指针、熟悉Python语法(赋值操作)

题目要求:

  • 给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法:双指针

class Solution:
    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
        """
        Do not return anything, modify nums1 in-place instead.
        """
        nums1_cpy = nums1[:m]  #切取前m个元素 
        nums1[:] = []

        num1Ptr = 0
        num2Ptr = 0

        while num1Ptr < m and num2Ptr < n:
            if nums1_cpy[num1Ptr]<  nums2[num2Ptr]:
                nums1.append(nums1_cpy[num1Ptr])
                num1Ptr += 1
            else:
                nums1.append(nums2[num2Ptr])
                num2Ptr += 1
        
        if num1Ptr < m:
            nums1[num1Ptr+num2Ptr:] = nums1_cpy[num1Ptr:]

        if num2Ptr < n:
            nums1[num1Ptr+num2Ptr:] = nums2[num2Ptr:]

总结:

  • 可以发现上面代码中有 nums1[:] = [] ,这里[:]的意思是将nums1地址所存储的列表清空。而函数执行完之后,其他代码访问改地址就是修改后的(类似于cpp中的参数引用传递),如此即刻满足题目要求 “在原地修改”。

  • 参考
    https://www.zhihu.com/question/54282837

发布了18 篇原创文章 · 获赞 0 · 访问量 490

猜你喜欢

转载自blog.csdn.net/m0_38139098/article/details/104433111