原题:
给定两个有序整数数组 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]
分析:首先可以通过暴力求解法,先将两个数组合并,之后对其排序即可,这样只需要一行代码就行
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)
效果:
解法2:
分析:通过双指针求解 将pointer1指向数组1的头部,再将pointer2指向数组2的头部,每次比较指针所指的数,选其中最小的;之后对相应的指针加1,直到到达末尾;最后将剩余未加的直接加入数组即可
代码:
def merge(self, nums1: list, m: int, nums2: list, n: int) -> list:
i = len(nums1)
j = len(nums2)
pointer1 = 0
pointer2 = 0
while (pointer1<m)&(pointer2<j):
if nums1[pointer1]>nums2[pointer2]:
self.append(nums2[pointer2])
pointer2 = pointer2+1
else:
self.append(nums1[pointer1])
pointer1 = pointer1+1
if pointer1<m:
for h in range(pointer1,m):
self.append(nums1[h])
if pointer2<j:
for h in range(pointer2,j):
self.append(nums2[h])
return self
nums1 = [1, 2, 3, 0, 0, 0]
nums2 = [2, 5, 6]
self = []
List = (merge(self, nums1, 3, nums2, 3))
for i in range(0, len(List)):
print(List[i])
效果: