最为简单直接的方法就是合并后再用sort()函数进行排序,但是这样的时间复杂度较高,为O(m+n)log(m+n)。
使用三个指针,p3指向nums1最后一个元素的位置,p1指向nums1的 m-1 的位置,p2指向数组nums2的 n-1 的位置。判断p1和p2位置的元素的大小,将大值放入p3的位置,而后p3-1,p1和p2指向元素大的指针相应左移,以此类推即可完成题目要求。
class Solution:
def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
p1 = m - 1
p2 = n - 1
p3 = m + n - 1
while p1 >= 0 and p2 >= 0:
if nums1[p1] > nums2[p2]:
nums1[p3] = nums1[p1]
p1 -= 1
else:
nums1[p3] = nums2[p2]
p2 -= 1
p3 -= 1
#如果p1<0,则将nums2中剩余的部分都放到nums1的最左边,如果p2<0,则不需要多余的操作
if p1 < 0:
nums1[:p2 + 1] = nums2[:p2 + 1]