合并两个有序数组
给定两个有序整数数组 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]
思路1:先合并再排序(此方法空间并不是面试官想看到的,最好别用)
- 时间复杂度较差,为O((n+m)log(n+m))
.sort()的时间复杂度为O(nlogn)
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:m+n] = nums2[0:n]
nums1.sort()
总结:
题目中没有明确要求禁止使用排序 sort时
我门都可以用 .sort()
但是在6大排序算法中 我明确的自己会实现 不能用sort()
思路2:双指针 / 从前往后(此方法空间用的多,不推荐)
-
一般而言,对于有序数组可以通过 双指针法 达到O(n + m)O(n+m)的时间复杂度。
-
最直接的算法实现是将指针p1 置为 nums1的开头, p2为 nums2的开头,在每一步将最小值放入输出数组中。
-
由于 nums1 是用于输出的数组,需要将nums1中的前m个元素放在其他地方,也就需要 O(m)O(m) 的空间复杂度。
#此代码不写了,就别看了,看我下面的最优代码
复杂度分析
- 时间复杂度 : O(n + m)O(n+m)。
- 空间复杂度 : O(m)O(m)。
思路三:双指针 / 从后往前(面试官最想看到的)
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.
"""
while m > 0 and n > 0:
if nums1[m-1] > nums2[n-1]:
nums1[m+n-1] = nums1[m-1]
m -= 1
else:
nums1[m+n-1] = nums2[n-1]
n -= 1
if n>0: # 若nums1完了,nums2还没完
nums1[:n] = nums2[:n] # 把剩下nums2加在最开始
复杂度分析
- 时间复杂度 : O(n + m)O(n+m)。
- 空间复杂度 : O(1)O(1)。