题目:
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 num1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
(这里要求算法代码不返回任何值,直接修改nums1列表)
解题思路
1.m肯定小于或者等于nums1的长度
2.只取nums1的前m个数的话,那么就是说nums[m]以及nums[m]以后的数都是可以舍弃的
3.反正nums1只要前m个数,那么后面的数先舍弃谁都一样,调用pop方法从最后一个删起
4.要删了nums1下标为m到len(nums1)的数,每次删一个,就需要删len(nums1)-m次.可以for循环range(len(nums1)-m)或者range(m,len(nums1)),每次循环都pop一下,就得到了“切片”后的nums1
5.剩下直接对nums2进行切片,然后再调用sort方法,ok
小白求教:
为什么我的代码最后面不能直接写成
nums1.extend(nums2[:n]).sort()
看了一下extend方法定义,是因为没有返回值所以不能这样使用吗?等一个大佬给我指点确认一下
代码
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.
"""
for p in range(m,len(nums1)): nums1.pop() #for循环这样写又少一行,好赖皮呀
nums1.extend(nums2[:n])
nums1.sort()
看到有别的大佬两行代码:
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[m:] = nums2[0:n]
nums1.sort()
不过面试官不会想看到上面的解法的
上leetcode看其他效率更高的题解吧