6. 合并排序数组 II
合并两个排序的整数数组A和B变成一个新的数组。
样例
给出A=[1,2,3,4],B=[2,4,5,6],返回 [1,2,2,3,4,4,5,6]
class Solution:
"""
@param A: sorted integer array A
@param B: sorted integer array B
@return: A new sorted integer array
"""
def mergeSortedArray(self, A, B):
# write your code here
result = []
i = 0
j = 0
while i<len(A) or j<len(B):
if i == len(A):
result.append(B[j])
j += 1
elif j == len(B):
result.append(A[i])
i += 1
elif A[i] < B[j]:
result.append(A[i])
i +=1
else:
result.append(B[j])
j+=1
return result