题目描述:
Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.
Note:
The number of elements initialized in nums1 and nums2 are m and n respectively.
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.
例子:
Example:
Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
Output: [1,2,2,3,5,6]
思路1:
将nums1后面的0全部替换成nums2里面的数,再将nums1调用sort函数。当然这样的解法面试官是肯定不能同意的。
思路2:
将指针m,n分别指向nums1的最后一个有效数,nums2的最后一个有效数。即nums1[m-1]和nums2[n-1]。比较大小,如果nums1大,则将这个数和nums1最后一个位置nums1[m+n-1]的数调换,并且m-1。否则,将nums2这个数赋值给nums1[m+n-1],并且n-1。直到m或者n为0停止。如果nums2有剩余,则将剩下的0到n-1个数全部赋值给nums1的0到n-1的数(因为剩下的位置全是0了)。
代码1:
class Solution:
def merge(self, nums1, m: int, nums2, n: int) -> None:
"""
Do not return anything, modify nums1 in-place instead.
"""
for i in range(n):
nums1[m+i] = nums2[i]
nums1.sort()
代码2:
class Solution:
def merge(self, nums1, m: int, nums2, n: int) -> None:
"""
Do not return anything, modify nums1 in-place instead.
"""
while n > 0 and m > 0:
if nums1[m-1] >= nums2[n-1]:
nums1[m-1], nums1[m+n-1] = nums1[m+n-1], nums1[m-1]
m -= 1
else:
nums1[m+n-1] = nums2[n-1]
n -= 1
if n > 0 and m == 0:
nums1[:n] = nums2[:n]
解释:
insert方法并不适用于此题,因为题目本身已经给你开拓了nums1需要被加入的位置。