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