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]
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 【自荐】一款简洁、开源的在线白板工具 Drawnix