写作目的:理解双指针、熟悉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中的参数引用传递),如此即刻满足题目要求 “在原地修改”。