python--lintcode6. 合并排序数组 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
        i,j=0,0
        result=[]
        while(i<len(A) and j<len(B)):
            if(A[i]<B[j]):
                result.append(A[i])
                i+=1
            else :
                result.append(B[j])
                j+=1
        result+=A[i:]
        result+=B[j:]
        return result




s=Solution()
print(s.mergeSortedArray([1,2],[4,5]))