LeetCode-88. 合并两个有序数组
给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。
想法:
- 做一个新数组nums3,长度和nums1相同,直接循环判断两个数组,将符合条件的放入到nums3中,nums1和nums2谁还有剩余,谁就全部赋值给nums3,最后需要将nums3中的值赋值给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 and n != 0:
for l in range(len(nums2)):
nums1[l] = nums2[l]
elif m != 0 and n!= 0:
nums3 = [0]*len(nums1)
i, j, k = 0, 0, 0
while i <= m-1 and j <= n-1:
if nums1[i] > nums2[j]:
nums3[k] = nums2[j]
k += 1
j += 1
else:
nums3[k] = nums1[i]
k += 1
i += 1
if i == m:
nums3[i+j:] = nums2[j:]
if j == n:
nums3[i+j:] = nums1[i:m]
for i in range(len(nums3)):
nums1[i] = nums3[i]
- 上面方法空间耗费大,所以改进一下方法,从nums1的末尾,也就是m+n位依次填入数据,最后剩余的再判断谁有剩余,统一赋值过去
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 and n != 0:
for l in range(len(nums2)):
nums1[l] = nums2[l]
elif m != 0 and n!= 0:
sumn = m + n
i, j = m, n
while i >= 1 and j >= 1:
if nums1[i-1] > nums2[j-1]:
nums1[sumn-1] = nums1[i-1]
i -= 1
sumn -= 1
else:
nums1[sumn-1] = nums2[j-1]
j -= 1
sumn -= 1
if i == 0:
nums1[0:sumn] = nums2[0:j]
if j == 0:
nums1[0:sumn] = nums1[0:i]