给定两个有序整数数组nums1和nums2,合并nums1和nums2为数组nums1,
注意:数组nums1和nums2初始化元素个数分别为m和n
假设num1有足够空间(长度超过m+n或与其相等)保存nums2中所有的元素
举例:Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
Output:
[1,2,2,3,5,6]
Python代码如下:
def merge(nums1, m, nums2, n): count = 0 # nums2索引 index = 0 # nums1索引 while count < n: # 判断条件(nums2长度) if nums1[index] > nums2[count]: # nums2中元素插入nums1中有效部分(非0部分) # for i in range(m+count,index,-1): #注释的两行和注释下面一行效果相同 # nums1[i] = nums1[i-1] nums1[index + 1:m + count + 1] = nums1[index:m + count] nums1[index] = nums2[count] count += 1 if index > m + count - 1: # nums2中元素插入nums1列表后面全0区域 nums1[index] = nums2[count] count += 1 index += 1 if __name__ == '__main__': nums1=[1,2,3,0,0,0] m=3 nums2=[2,5,6] n=3 merge(nums1,m,nums2,n) print(nums1)输入结果如下:
[1, 2, 2, 3, 5, 6]