88.2つの順序付けられた配列を組み合わせる
2つの順序付き整数配列nums1とnums2がある場合、nums2をnums1にマージして、nums1を順序付き配列にしてください。
説明:
nums1とnums2の要素数をそれぞれmとnに初期化します。
nums1には、要素をnums2に格納するのに十分なスペース(スペースのサイズが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.
"""
tem_list = nums1[:m] + nums2
nums1[:] = [] #需要注意是这里,看清题目,返回的是nums1,所以这里赋值一下
start_1, start_2 = 0,m
while start_1 < m or start_2 < m+n:
if start_1 == m:
nums1.append(tem_list[start_2])
start_2+=1
elif start_2 == m+n:
nums1.append(tem_list[start_1])
start_1+=1
elif tem_list[start_1] <= tem_list[start_2]:
nums1.append(tem_list[start_1])
start_1+=1
else:
nums1.append(tem_list[start_2])
start_2+=1
(書いた後は複雑に感じます)、2つの配列を直接追加する必要がないため、スペースの複雑さを軽減できます。
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.
"""
tem = nums1[:m]
nums1[:] = []
i,j = 0, 0
while i < m or j < n:
if i == m:
nums1.append(nums2[j])
j+=1
elif j == n:
nums1.append(tem[i])
i+=1
elif tem[i] <= nums2[j]:
nums1.append(tem[i])
i+=1
else:
nums1.append(nums2[j])
j+=1