给定两个有序整数数组 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]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
不用内置函数的话:大致思路是把nums1复制一下给nums3,然后遍历nums2和nums3中的所有元素,遍历中间用if语句来比较,哪个小就哪个放进去。尝试过先合并再把小的往前移,但空间复杂度大了,不好。
class Solution:
def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
nums3=nums1[:]#复制nums1,之前试过nums3=nums1,结果发现这样就让两个数组内存地址一样了,这样在排序的过程中nums3就会跟着nums1一起改变,不行
a,b,c=0,0,0
while a<m and b<n:
if nums3[a]<nums2[b]:
nums1[c]=nums3[a]
a+=1
c+=1
else:
nums1[c]=nums2[b]
b+=1
c+=1
if a==m:
while b<n:
nums1[c]=nums2[b]
b+=1
c+=1
else:
while a<m:
nums1[c]=nums3[a]
a+=1
c+=1
return nums1
使用sorted函数:
class Solution:
def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
nums1[:]=sorted(nums1[:m]+nums2)
一行就无了。。。怎么感觉贱贱的哈哈