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[:] = sorted(nums1[:m] + nums2)
#从后往前排序,先确定k-1个位置的元素
k = m + n - 1
while m > 0 and n > 0:
if nums1[m - 1] > nums2[n - 1]:
nums1[k] = nums1[m - 1]
m -= 1
else:
nums1[k] = nums2[n - 1]
n -= 1
k -= 1
#m==0的时候有可能n!=0,因为需要将剩余的n个元素赋值到nums1中去
#n==0的时候有可能m!=0,那就nums1中的元素已经是排好序的不用动即可
nums1[:n] = nums2[:n]
#双指针
#p:指向nums1的指针
#q:指向nums2的指针
sorted = []
p1, p2 = 0, 0
while p1 < m or p2 < n:
if p1 == m:
sorted.append(nums2[p2])
p2 += 1
elif p2 == n:
sorted.append(nums1[p1])
p1 += 1
elif nums1[p1] < nums2[p2]:
sorted.append(nums1[p1])
p1 += 1
else:
sorted.append(nums2[p2])
p2 += 1
nums1[:] = sorted
解题分析
- 从后往前找,不需要借助额外空间
- 因为num1的长度是m+n,nums2的长度只有n
- nums1和nums2都是已经排序好的数组
- 每次比较最末尾的两个元素,然后放在m+n-1的位置上即可
- 最终m==0,那么n可能不是0,需要将个元素都赋值在nums1中
- 最终n==0,那么m可能不为0,但是都是已经排序好的,不需要在进行参与排序
- 双指针,从前往后找,需要借助额外空间
- 设定两个指针,分别指向两个数组的头部
- 核心思想就是每次找到两个数组最小的元素,然后添加到结果数组中去
- 如果其中p1==m或者p2==n;那么就遍历另一个数组直接添加即可