88. 合并两个有序数组
class No_0088_Merge:
"""
题干:
https://leetcode.cn/problems/merge-sorted-array/
两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目
合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列
题目中要求合并后数组不应由函数返回,而是存储在数组nums1中, nums1预留了足够的0
解析:
两个数组分别用一个指针记录当前位置,比较指针位置的值,输出后指针向后移动。
例如:A=[*1,3,5], B=[*2,6,8], *号代表指针
1<2,输出:1, 状态:[1,*3,5], [*2,6,8]
2<3,输出:2, 状态:[1,*3,5], [2,*6,8]
3<6,输出:3, 状态:[1,3,*5], [2,*6,8]
5<6,输出:5, 状态:[1,3,5], [2,*6,8]
A已经完成,顺序输出B的剩余元素
"""
@staticmethod
def sol_two_points(nums1: list, m: int, nums2: list, n: int):
p0, p1, p2 = 0, 0, 0
_nums1 = nums1.copy()
while p1 < m and p2 < n:
if _nums1[p1] <= nums2[p2]:
nums1[p0] = _nums1[p1]
p1 += 1
else:
nums1[p0] = nums2[p2]
p2 += 1
p0 += 1
if p1 < m:
nums1[p0:] = _nums1[p1:m]
elif p2 < n:
nums1[p0:] = nums2[p2:n]
return nums1
@staticmethod
def sol_two_points_rev(nums1: list, m: int, nums2: list, n: int):
p0, p1, p2 = m+n-1, m-1, n-1
while p1 >= 0 and p2 >= 0:
if nums1[p1] >= nums2[p2]:
nums1[p0] = nums1[p1]
p1 -= 1
else:
nums1[p0] = nums2[p2]
p2 -= 1
p0 -= 1
if p1 < 0:
nums1[:p0+1] = nums2[:p2+1]
elif p2 < 0:
nums1[:p0+1] = nums1[:p1+1]
return nums1