解题思路:
这个题目有一个比较好的地方就是,nums1的长度正好等于n + m,所以可以利用这一点;
1.先把nums1的前m的数字移到最后面去,腾位置;
2.然后将移到后面的数和nums2一一比较,将较小的放到nums1的前面去,经过O(n + m)次判断,就可以将nums1和nums2有序的放入nums1中了。
具体代码如下:
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.
"""
if m == 0:
for i in range(n):
nums1[i] = nums2[i]
elif n == 0:
pass
else:
nums1[-m:] = nums1[:m]
first_ptr = n #[n, n + m -1]
second_ptr = 0 #[0, n - 1]
ptr = 0
while first_ptr < n+m and second_ptr < n:
if nums1[first_ptr] < nums2[second_ptr]:
nums1[ptr] = nums1[first_ptr]
ptr += 1
first_ptr += 1
elif nums1[first_ptr] >= nums2[second_ptr]:
nums1[ptr] = nums2[second_ptr]
ptr += 1
second_ptr += 1
while first_ptr == n+m and second_ptr < n:
nums1[ptr] = nums2[second_ptr]
ptr += 1
second_ptr += 1
while first_ptr < n+m and second_ptr == n:
nums1[ptr] = nums1[first_ptr]
ptr+=1
first_ptr += 1