88 合并两个有序数组
给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。
设置3个指针,一个指向nums1的m,一个指向nums2的n,一个指向nums的末尾,m+n-1。然后从末尾依次对比m-1和n-1指针对应的元素值,大的放在nums1末尾处,不断更新指针,需要注意的是if之后只能写else,不然两个if的都可能循环,但是k只更新了一次,最后,m和n肯定有一个先全部遍历完,这个时候用nums1[:n]=nums2[:n],因为nums1比nums2要长,所以,如果是m先到0的话,就是把剩下nums2的全部都接到nums前面去 ,如果n先到0的话,这个意思就是保持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.
"""
k = m+n-1
while m > 0 and n > 0:
if nums1[m - 1] > nums2[n - 1]:
nums1[k] = nums1[m - 1]
m -= 1
else:
nums1[k] = nums2[n - 1]
n -= 1
k -= 1
nums1[:n] = nums2[:n]