88. 合并两个有序数组
https://leetcode-cn.com/problems/merge-sorted-array/
难度:简单
题目:
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。
示例:
示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
示例 2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
分析
如果只是为了刷题AC,那么两行代码就够了。
但如果是为了练手,最好还是用双指针的思路去做这道题:
- 现将nums2合并入nums1
- 左右指针分别指向0和m的位置
- 当左指针小于右指针,左指针+1
- 当左指针小于右指针
- 先用tmp存储右指针的值
- 然后 nums1[left:right] = nums1[left+1:right+1]
- 最后将nums1[left] = tmp 完成滚动更新
- 然后左右指针同时+1
- 判断结束条件为 左指针<右指针< 两个数组总长m+n
解题1:快速的通关:
class Solution:
def merge(self, nums1, m, nums2, n):
"""
Do not return anything, modify nums1 in-place instead.
"""
nums1[m:] = nums2
nums1.sort()
解题2:双指针滚动更新:
class Solution:
def merge(self, nums1, m, nums2, n):
"""
Do not return anything, modify nums1 in-place instead.
"""
nums1[m:] = nums2
left = 0
right = m
while left <= right < m + n:
if nums1[left] > nums1[right]:
tmp = nums1[right]
nums1[left + 1:right + 1] = nums1[left:right]
nums1[left] = tmp
left += 1
right += 1
else:
left += 1
欢迎关注我的公众号: 清风Python,带你每日学习Python算法刷题的同时,了解更多python小知识。
有喜欢力扣刷题的小伙伴可以加我微信(King_Uranus)互相鼓励,共同进步,一起玩转超级码力!
我的个人博客:https://qingfengpython.cn