说一个比较偷懒的方法,即使用python自带的sort()函数,一步到位
class Solution: def merge(self , A, m, B, n): # write code here for i in range(n): A[m+i]=B[i] return A.sort()
如果不用sort()来排序的话,此题考的明显是归并排序的“归并”部分,即分治法的“治”部分。
思路为:先从两个序列的后面开始比较,用两个指针,将大的放在A的最后,以此类推一直到最后比较完了,直接将B剩余的值复制到
A即可。
# # # @param A int整型一维数组 # @param B int整型一维数组 # @return void # class Solution: def merge(self , A, m, B, n): # write code here # temp = [] i,j,k=m-1,n-1,m+n-1 while i>=0 and j>=0: if A[i]>B[j]: A[k] = A[i] k -= 1 i -= 1 else: A[k] = B[j] k -= 1 j -= 1 A[:j+1] = B[:j+1] return A