给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,
nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。
nums2 的长度为 n 。
import numpy as np
class Solution(object):
def merge(self, nums1, m, nums2, n):
"""
:type nums1: List[int]
:type m: int
:type nums2: List[int]
:type n: int
:rtype: None Do not return anything, modify nums1 in-place instead.
"""
t = 0
if n != 0:
for i in range(n):
if m == 0:
for i in range(m+n):
nums1.pop()
for j in range(n):
nums1.append(nums2[j])
break
else:
for j in range(m+n):
if nums2[i] <= nums1[j]:
nums1.insert(j, nums2[i])
nums1.pop()
break
if nums2[i] >= max(nums1):
if nums2[i] == 0:
t += 1
nums1.insert(int(np.flatnonzero(np.array(nums1))[-1]+1)+t, nums2[i])
if nums2[i] != 0:
t = 0
nums1.pop()
break
if __name__ == '__main__':
# case4
# nums1 = [-1,0,1,1,0,0,0,0,0]
# nums2 = [-1,0,2,2,3]
# nums1 = [1,2,3,0,0,0]
# nums2 = [2,5,6]
nums1 = [-12,0,0,0,0,0]
nums2 = [-1,0,0,0,2]
n = len(nums2)
m = len(nums1) - n
Solution = Solution()
Solution.merge(nums1, m, nums2, n)
print(nums1)